stable-cascade

Running

multimodalart HF staff commited on Feb 12

Commit

f852ba6

•

1 Parent(s): 7023c2f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,7 @@ if torch.cuda.is_available():
     if PREVIEW_IMAGES:
         previewer = Previewer()
-        previewer.load_state_dict(torch.load("previewer/previewer_v1_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):