Spaces:

leowajda
/

diffusion_model

Paused

leowajda commited on Dec 30, 2023

Commit

7195a47

1 Parent(s): fc74cfd

cache models

Files changed (1) hide show

app.py CHANGED Viewed

@@ -81,15 +81,15 @@ gallery = gr.Gallery(
 diffusion_models = {
     "linear":
         DiffusionSampler(
-            model=from_pretrained_keras("leowajda/linear_diffusion"),
-            ema_model=from_pretrained_keras("leowajda/linear_diffusion_ema"),
             noise_scheduler="linear"
         ),
     "cosine":
         DiffusionSampler(
-            model=from_pretrained_keras("leowajda/cosine_diffusion"),
-            ema_model=from_pretrained_keras("leowajda/cosine_diffusion_ema"),
             noise_scheduler="cosine"
         )
 }

 diffusion_models = {
     "linear":
         DiffusionSampler(
+            model=from_pretrained_keras("leowajda/linear_diffusion", cache_dir="cache"),
+            ema_model=from_pretrained_keras("leowajda/linear_diffusion_ema", cache_dir="cache"),
             noise_scheduler="linear"
         ),
     "cosine":
         DiffusionSampler(
+            model=from_pretrained_keras("leowajda/cosine_diffusion", cache_dir="cache"),
+            ema_model=from_pretrained_keras("leowajda/cosine_diffusion_ema", cache_dir="cache"),
             noise_scheduler="cosine"
         )
 }