Wuerstchen-preload

Paused

dome272 commited on Sep 10, 2023

Commit

4da08fe

•

1 Parent(s): 852c3ff

fix filename

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,7 +39,7 @@ if torch.cuda.is_available():
     if PREVIEW_IMAGES:
         previewer = Previewer()
-        previewer.load_state_dict(torch.load(r"C:\Users\d6582\Documents\ml\wuerstchen\diffusers\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):