bingliangzhang00

/

STeP-mri

like 1

• "torch.nn.modules.linear.Linear",
• "omegaconf.nodes.AnyNode",
• "collections.defaultdict",
• "diffusers.models.autoencoders.vae.Encoder",
• "torch.nn.modules.activation.SiLU",
• "typing.Any",
• "omegaconf.dictconfig.DictConfig",
• "omegaconf.base.Metadata",
• "torch.FloatStorage",
• "__builtin__.set",
• "diffusers.models.downsampling.Downsample2D",
• "diffusers.models.upsampling.Upsample2D",
• "diffusers.models.resnet.ResnetBlock2D",
• "__builtin__.long",
• "torch._utils._rebuild_tensor_v2",
• "torch.nn.modules.container.ModuleList",
• "torch._utils._rebuild_parameter",
• "__builtin__.dict",
• "torch.nn.modules.normalization.GroupNorm",
• "torch.nn.modules.dropout.Dropout",
• "collections.OrderedDict",
• "diffusers.models.attention_processor.AttnProcessor2_0",
• "diffusers.models.autoencoders.autoencoder_kl.AutoencoderKL",
• "diffusers.configuration_utils.FrozenDict",
• "torch.nn.modules.conv.Conv2d",
• "diffusers.models.attention_processor.Attention",
• "diffusers.models.autoencoders.vae.Decoder",
• "diffusers.models.unets.unet_2d_blocks.DownEncoderBlock2D",
• "__builtin__.list",
• "diffusers.models.unets.unet_2d_blocks.UpDecoderBlock2D",
• "omegaconf.listconfig.ListConfig",
• "diffusers.models.unets.unet_2d_blocks.UNetMidBlock2D",
• "omegaconf.base.ContainerMetadata"

How to fix it?

• "torch.FloatStorage",
• "omegaconf.base.Metadata",
• "torch.nn.modules.linear.Linear",
• "typing.Any",
• "torch.nn.modules.dropout.Dropout",
• "diffusers.models.attention.BasicTransformerBlock",
• "__builtin__.set",
• "diffusers.models.unets.unet_3d_blocks.UpBlockSpatioTemporal",
• "diffusers.models.upsampling.Upsample2D",
• "torch.nn.modules.normalization.LayerNorm",
• "__builtin__.list",
• "diffusers.models.attention.FeedForward",
• "diffusers.models.activations.GEGLU",
• "omegaconf.listconfig.ListConfig",
• "diffusers.models.resnet.AlphaBlender",
• "diffusers.models.resnet.SpatioTemporalResBlock",
• "torch.nn.modules.normalization.GroupNorm",
• "omegaconf.base.ContainerMetadata",
• "diffusers.models.downsampling.Downsample2D",
• "diffusers.configuration_utils.FrozenDict",
• "diffusers.models.unets.unet_3d_blocks.DownBlockSpatioTemporal",
• "diffusers.models.embeddings.TimestepEmbedding",
• "collections.defaultdict",
• "__builtin__.dict",
• "omegaconf.nodes.AnyNode",
• "diffusers.models.attention.TemporalBasicTransformerBlock",
• "torch._utils._rebuild_parameter",
• "diffusers.models.resnet.TemporalResnetBlock",
• "diffusers.models.unets.unet_3d_blocks.UNetMidBlockSpatioTemporal",
• "model.st_model.UNetSpatioTemporalDiffusion",
• "diffusers.models.transformers.transformer_temporal.TransformerSpatioTemporalModel",
• "torch.nn.modules.activation.SiLU",
• "torch.nn.modules.container.ModuleList",
• "torch._utils._rebuild_tensor_v2",
• "diffusers.models.resnet.ResnetBlock2D",
• "__builtin__.long",
• "omegaconf.dictconfig.DictConfig",
• "diffusers.models.embeddings.Timesteps",
• "diffusers.models.attention_processor.Attention",
• "collections.OrderedDict",
• "torch.nn.modules.conv.Conv2d",
• "torch.nn.modules.conv.Conv3d",
• "diffusers.models.attention_processor.AttnProcessor2_0"

How to fix it?

• "torch.LongStorage",
• "collections.OrderedDict",
• "diffusers.configuration_utils.FrozenDict",
• "diffusers.schedulers.scheduling_ddpm.DDPMScheduler",
• "torch.FloatStorage",
• "torch._utils._rebuild_tensor_v2"

How to fix it?