stable-cascade

Running

cocktailpeanut commited on Feb 24, 2024

Commit

f4f5727

1 Parent(s): a5c3e20

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -92,7 +92,8 @@ def generate(
     num_images_per_prompt: int = 2,
 #    profile: gr.OAuthProfile | None = None,
 ) -> PIL.Image.Image:
-    previewer.eval().requires_grad_(False).to(device).to(dtype)
     prior_pipeline.to(device)
     decoder_pipeline.to(device)

     num_images_per_prompt: int = 2,
 #    profile: gr.OAuthProfile | None = None,
 ) -> PIL.Image.Image:
+    if previewer != None:
+        previewer.eval().requires_grad_(False).to(device).to(dtype)
     prior_pipeline.to(device)
     decoder_pipeline.to(device)