UT-Austin-PML

/

SiDA-StableDiffusion

like 0

Follow

Probabilistic Machine Learning at UT-Austin 18

• "torch.nn.modules.linear.Linear",
• "torch.nn.modules.normalization.LayerNorm",
• "diffusers.models.attention_processor.Attention",
• "diffusers.models.transformers.sana_transformer.SanaTransformerBlock",
• "diffusers.models.embeddings.PatchEmbed",
• "diffusers.models.transformers.sana_transformer.SanaModulatedNorm",
• "torch.storage._load_from_bytes",
• "diffusers.models.transformers.sana_transformer.SanaTransformer2DModel",
• "diffusers.configuration_utils.FrozenDict",
• "torch._utils._rebuild_tensor_v2",
• "diffusers.models.transformers.sana_transformer.GLUMBConv",
• "diffusers.models.embeddings.PixArtAlphaCombinedTimestepSizeEmbeddings",
• "diffusers.models.embeddings.TimestepEmbedding",
• "diffusers.models.attention_processor.SanaLinearAttnProcessor2_0",
• "diffusers.models.transformers.sana_transformer.SanaAttnProcessor2_0",
• "torch.nn.modules.activation.SiLU",
• "torch.nn.modules.container.ModuleList",
• "diffusers.models.embeddings.Timesteps",
• "torch.nn.modules.conv.Conv2d",
• "torch.nn.modules.dropout.Dropout",
• "diffusers.models.normalization.RMSNorm",
• "torch.nn.modules.activation.GELU",
• "collections.OrderedDict",
• "torch._utils._rebuild_parameter",
• "diffusers.models.embeddings.PixArtAlphaTextProjection",
• "torch.Size",
• "diffusers.models.normalization.AdaLayerNormSingle"

How to fix it?