stable-cascade

Running

cocktailpeanut commited on Feb 26, 2024

Commit

3f80461

1 Parent(s): f4f5727

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -92,8 +92,10 @@ def generate(
     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)

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