Spaces:
Sleeping
Sleeping
Commit
•
f852ba6
1
Parent(s):
7023c2f
Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ if torch.cuda.is_available():
|
|
45 |
|
46 |
if PREVIEW_IMAGES:
|
47 |
previewer = Previewer()
|
48 |
-
previewer.load_state_dict(torch.load("previewer/
|
49 |
previewer.eval().requires_grad_(False).to(device).to(dtype)
|
50 |
|
51 |
def callback_prior(i, t, latents):
|
|
|
45 |
|
46 |
if PREVIEW_IMAGES:
|
47 |
previewer = Previewer()
|
48 |
+
previewer.load_state_dict(torch.load("previewer/text2img_wurstchen_b_v1_previewer_100k.pt")["state_dict"])
|
49 |
previewer.eval().requires_grad_(False).to(device).to(dtype)
|
50 |
|
51 |
def callback_prior(i, t, latents):
|