zhangyang-0123

/

EcoDiffPrunedModels

like 2

• "diffusers.configuration_utils.FrozenDict",
• "diffusers.models.upsampling.Upsample2D",
• "diffusers.models.embeddings.Timesteps",
• "diffusers.models.downsampling.Downsample2D",
• "__builtin__.set",
• "utils.SkipConnection",
• "torch.BFloat16Storage",
• "collections.OrderedDict",
• "diffusers.models.unets.unet_2d_blocks.UpBlock2D",
• "diffusers.models.activations.GEGLU",
• "torch.nn.modules.linear.Linear",
• "diffusers.models.attention_processor.AttnProcessor2_0",
• "torch.nn.modules.dropout.Dropout",
• "diffusers.models.attention.BasicTransformerBlock",
• "torch._utils._rebuild_parameter",
• "diffusers.models.embeddings.TimestepEmbedding",
• "torch.nn.modules.conv.Conv2d",
• "diffusers.models.unets.unet_2d_condition.UNet2DConditionModel",
• "torch.nn.modules.normalization.LayerNorm",
• "diffusers.models.unets.unet_2d_blocks.CrossAttnUpBlock2D",
• "diffusers.models.unets.unet_2d_blocks.UNetMidBlock2DCrossAttn",
• "diffusers.models.unets.unet_2d_blocks.CrossAttnDownBlock2D",
• "torch.nn.modules.container.ModuleList",
• "diffusers.models.transformers.transformer_2d.Transformer2DModel",
• "torch.nn.modules.activation.SiLU",
• "diffusers.models.attention_processor.Attention",
• "torch.nn.modules.normalization.GroupNorm",
• "diffusers.models.unets.unet_2d_blocks.DownBlock2D",
• "diffusers.models.attention.FeedForward",
• "torch._utils._rebuild_tensor_v2",
• "diffusers.models.resnet.ResnetBlock2D"

How to fix it?