PrunaAI

/

FLUX.1-schnell-4bit

like 8

• "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?