add fastcore and diffusers

Author: tcapelle

cloud_diffusion/version.py

@@ -1 +1 @@
-__version__ = "0.3.0"
+__version__ = "0.3.1"

inference.py

@@ -10,8 +10,8 @@
 
 
 from cloud_diffusion.dataset import download_dataset, CloudDataset
-from cloud_diffusion.ddpm import UNet2D, get_unet_params
-from cloud_diffusion.utils import ddim_sampler, parse_args, set_seed, ls
+from cloud_diffusion.ddpm import UNet2D, get_unet_params, ddim_sampler
+from cloud_diffusion.utils import parse_args, set_seed, ls
 from cloud_diffusion.wandb import log_images, save_model, to_video, vhtile
 
 PROJECT_NAME = "ddpm_clouds"

setup.py

@@ -20,11 +20,12 @@
     ],
     install_requires=[
         "torch",
-        "fastcore",
         "fastprogress",
+        "fastcore",
         "wandb",
         "numpy",
         "matplotlib",
+        "diffusers",
     ],
     classifiers=[
         "Development Status :: 4 - Beta",