Hugging Face's logo

Spaces:
warp-ai
/
Wuerstchen
Runtime error

App Files Community
30
Wauplin HF Staff commited on
Commit
1b7bc92
·
1 Parent(s): 9731d82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -44,11 +44,7 @@ if torch.cuda.is_available():
44
 
45
  if PREVIEW_IMAGES:
46
  previewer = Previewer()
47
- previewer.load_state_dict(
48
- torch.load("previewer/text2img_wurstchen_b_v1_previewer_100k.pt")[
49
- "state_dict"
50
- ]
51
- )
52
  previewer.eval().requires_grad_(False).to(device).to(dtype)
53
 
54
  def callback_prior(i, t, latents):
 
44
 
45
  if PREVIEW_IMAGES:
46
  previewer = Previewer()
47
+ previewer.load_state_dict(torch.load("previewer/text2img_wurstchen_b_v1_previewer_100k.pt")["state_dict"])
 
 
 
 
48
  previewer.eval().requires_grad_(False).to(device).to(dtype)
49
 
50
  def callback_prior(i, t, latents):