PrunaAI

/

FLUX.1-schnell-4bit

like 8

• "transformers.models.t5.modeling_t5.T5Attention",
• "torch.float8_e4m3fn",
• "torch._tensor._rebuild_from_type_v2",
• "transformers.models.t5.configuration_t5.T5Config",
• "collections.OrderedDict",
• "transformers.models.t5.modeling_t5.T5LayerNorm",
• "torch.nn.modules.sparse.Embedding",
• "transformers.models.t5.modeling_t5.T5LayerSelfAttention",
• "transformers.models.t5.modeling_t5.T5LayerFF",
• "optimum.quanto.tensor.qbytes.QBytesTensor",
• "torch.FloatStorage",
• "torch.bfloat16",
• "torch.nn.modules.container.ModuleList",
• "torch.device",
• "transformers.models.t5.modeling_t5.T5Stack",
• "transformers.models.t5.modeling_t5.T5Block",
• "torch.nn.modules.dropout.Dropout",
• "transformers.models.t5.modeling_t5.T5DenseGatedActDense",
• "torch._utils._rebuild_tensor_v3",
• "torch.serialization._get_layout",
• "torch._utils._rebuild_tensor_v2",
• "torch._utils._rebuild_wrapper_subclass",
• "torch.storage.UntypedStorage",
• "torch.BFloat16Storage",
• "transformers.models.t5.modeling_t5.T5EncoderModel",
• "optimum.quanto.tensor.qtype.qtype",
• "optimum.quanto.nn.qlinear.QLinear",
• "torch._utils._rebuild_parameter",
• "transformers.activations.NewGELUActivation",
• "__builtin__.set"

How to fix it?

• "torch.device",
• "optimum.quanto.tensor.qbits.tinygemm.packed.TinyGemmPackedTensor",
• "optimum.quanto.tensor.qtype.qtype",
• "torch.FloatStorage",
• "__builtin__.set",
• "torch.bfloat16",
• "diffusers.models.normalization.AdaLayerNormContinuous",
• "diffusers.models.attention_processor.Attention",
• "torch._utils._rebuild_parameter",
• "diffusers.models.transformers.transformer_flux.FluxTransformer2DModel",
• "torch._utils._rebuild_wrapper_subclass",
• "diffusers.models.normalization.AdaLayerNormZero",
• "diffusers.models.transformers.transformer_flux.FluxTransformerBlock",
• "torch.nn.modules.activation.GELU",
• "torch.BFloat16Storage",
• "torch.uint8",
• "torch.serialization._get_layout",
• "torch.nn.modules.container.ModuleList",
• "diffusers.models.embeddings.PixArtAlphaTextProjection",
• "diffusers.models.transformers.transformer_flux.EmbedND",
• "optimum.quanto.tensor.qbits.tinygemm.qbits.TinyGemmQBitsTensor",
• "diffusers.models.normalization.RMSNorm",
• "diffusers.models.embeddings.Timesteps",
• "diffusers.models.activations.GELU",
• "torch.nn.modules.dropout.Dropout",
• "diffusers.models.embeddings.TimestepEmbedding",
• "torch.Size",
• "torch.IntStorage",
• "diffusers.models.embeddings.CombinedTimestepTextProjEmbeddings",
• "diffusers.models.normalization.AdaLayerNormZeroSingle",
• "diffusers.configuration_utils.FrozenDict",
• "optimum.quanto.nn.qlinear.QLinear",
• "torch.nn.modules.activation.SiLU",
• "diffusers.models.attention.FeedForward",
• "diffusers.models.transformers.transformer_flux.FluxSingleTransformerBlock",
• "diffusers.models.attention_processor.FluxAttnProcessor2_0",
• "torch.nn.modules.linear.Linear",
• "collections.OrderedDict",
• "torch.nn.modules.normalization.LayerNorm",
• "torch._tensor._rebuild_from_type_v2",
• "torch.int8",
• "torch._utils._rebuild_tensor_v2",
• "diffusers.models.attention_processor.FluxSingleAttnProcessor2_0"

How to fix it?