Spaces:

warp-ai
/

Wuerstchen

Runtime error

Wauplin HF staff commited on Sep 14, 2023

Commit

1b7bc92

•

1 Parent(s): 9731d82

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,11 +44,7 @@ if torch.cuda.is_available():
     if PREVIEW_IMAGES:
         previewer = Previewer()
-        previewer.load_state_dict(
-            torch.load("previewer/text2img_wurstchen_b_v1_previewer_100k.pt")[
-                "state_dict"
-            ]
-        )
         previewer.eval().requires_grad_(False).to(device).to(dtype)
         def callback_prior(i, t, latents):

     if PREVIEW_IMAGES:
         previewer = Previewer()
+        previewer.load_state_dict(torch.load("previewer/text2img_wurstchen_b_v1_previewer_100k.pt")["state_dict"])
         previewer.eval().requires_grad_(False).to(device).to(dtype)
         def callback_prior(i, t, latents):