Spaces:

leowajda
/

diffusion_model

Paused

leowajda commited on Dec 29, 2023

Commit

d7af000

1 Parent(s): 7578496

fix noise_scheduler

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ gallery = gr.Gallery(
 linear_diffusion_model = DiffusionSampler(
     model=from_pretrained_keras("leowajda/linear_diffusion"),
     ema_model=from_pretrained_keras("leowajda/linear_diffusion_ema"),
-    noise_scheduler="cosine",
 )
 cosine_diffusion_model = DiffusionSampler(

 linear_diffusion_model = DiffusionSampler(
     model=from_pretrained_keras("leowajda/linear_diffusion"),
     ema_model=from_pretrained_keras("leowajda/linear_diffusion_ema"),
+    noise_scheduler="linear",
 )
 cosine_diffusion_model = DiffusionSampler(