multimodalart HF staff commited on
Commit
c1e1bbb
1 Parent(s): 52ef426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,7 +41,8 @@ if torch.cuda.is_available():
41
  prior_pipeline.prior = torch.compile(prior_pipeline.prior, mode="reduce-overhead", fullgraph=True)
42
  decoder_pipeline.decoder = torch.compile(decoder_pipeline.decoder, mode="reduce-overhead", fullgraph=True)
43
 
44
- #if PREVIEW_IMAGES:
 
45
  # previewer = Previewer()
46
  # previewer.load_state_dict(torch.load("previewer/text2img_wurstchen_b_v1_previewer_100k.pt")["state_dict"])
47
  # previewer.eval().requires_grad_(False).to(device).to(dtype)
 
41
  prior_pipeline.prior = torch.compile(prior_pipeline.prior, mode="reduce-overhead", fullgraph=True)
42
  decoder_pipeline.decoder = torch.compile(decoder_pipeline.decoder, mode="reduce-overhead", fullgraph=True)
43
 
44
+ if PREVIEW_IMAGES:
45
+ pass
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)