upload

Author: patrickvonplaten

examples/sample_loop.py

@@ -1,16 +1,13 @@
 #!/usr/bin/env python3
-import tempfile
-import sys
 import os
 import pathlib
 from modeling_ddpm import DDPM
 import PIL.Image
 import numpy as np
 
-model_ids = ["ddpm-lsun-cat", "ddpm-lsun-cat-ema", "ddpm-lsun-church-ema", "ddpm-lsun-church", "ddpm-lsun-bedroom", "ddpm-lsun-bedroom-ema", "ddpm-cifar10-ema", "ddpm-lsun-cifar10", "ddpm-lsun-celeba-hq", "ddpm-lsun-celeba-hq-ema"]
+model_ids = ["ddpm-lsun-cat", "ddpm-lsun-cat-ema", "ddpm-lsun-church-ema", "ddpm-lsun-church", "ddpm-lsun-bedroom", "ddpm-lsun-bedroom-ema", "ddpm-cifar10-ema", "ddpm-cifar10", "ddpm-celeba-hq", "ddpm-celeba-hq-ema"]
 
 for model_id in model_ids:
-
     path = os.path.join("/home/patrick/images/hf", model_id)
     pathlib.Path(path).mkdir(parents=True, exist_ok=True)