Spaces:

vivjay30
/

cdim

Running on Zero

VIVEK JAYARAM commited on Oct 24, 2024

Commit

83bba75

1 Parent(s): 83e40e0

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def process_image(image_choice, noise_sigma, operator_key, T, K):
     if model is None:
         model_type = "diffusers"
         model_name = "google/ddpm-celebahq-256" if "Celeb" in image_choice else "google/ddpm-curch-256"
-        model = DiffusionPipeline.from_pretrained("google/ddpm-celebahq-256").to(device).unet
         ddim_scheduler = DDIMScheduler(
             num_train_timesteps=1000,
             beta_start=0.0001,

     if model is None:
         model_type = "diffusers"
         model_name = "google/ddpm-celebahq-256" if "Celeb" in image_choice else "google/ddpm-curch-256"
+        model = DiffusionPipeline.from_pretrained(model_name).to(device).unet
         ddim_scheduler = DDIMScheduler(
             num_train_timesteps=1000,
             beta_start=0.0001,