pinning to old libs

Author: Thomas Capelle

inference.py

@@ -5,7 +5,8 @@
 from fastprogress import progress_bar
 
 from cloud_diffusion.dataset import download_dataset, CloudDataset
-from cloud_diffusion.ddpm import UNet2D, get_unet_params, ddim_sampler
+from cloud_diffusion.ddpm import ddim_sampler
+from cloud_diffusion.models import UNet2D, get_unet_params
 from cloud_diffusion.utils import parse_args, set_seed
 from cloud_diffusion.wandb import to_video, vhtile
 

setup.py

@@ -25,8 +25,8 @@
         "wandb",
         "numpy",
         "matplotlib",
-        "diffusers",
-        "denoising_diffusion_pytorch",
+        "diffusers==0.9.0",
+        "denoising_diffusion_pytorch==1.1.0",
     ],
     classifiers=[
         "Development Status :: 4 - Beta",