add eye diffusion
fec8e67
last.ckpt
Detected Pickle imports (11)
- "torch.nn.modules.loss.L1Loss",
- "collections.OrderedDict",
- "medical_diffusion.models.estimators.unet2.UNet",
- "medical_diffusion.models.noise_schedulers.gaussian_scheduler.GaussianNoiseScheduler",
- "medical_diffusion.models.embedders.time_embedder.TimeEmbbeding",
- "torch.FloatStorage",
- "medical_diffusion.models.embedders.cond_embedders.LabelEmbedder",
- "torch._utils._rebuild_tensor_v2",
- "torch.LongStorage",
- "medical_diffusion.models.embedders.latent_embedders.VAE",
- "torch.optim.adamw.AdamW"
How to fix it?
2.44 GB
add eye diffusion