multimodalart HF staff commited on
Commit
2e6feb3
1 Parent(s): 03fb31d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ if torch.cuda.is_available():
39
  decoder_pipeline.to(device)
40
 
41
  if USE_TORCH_COMPILE:
42
- prior_pipeline.prior = torch.compile(prior_pipeline.prior)
43
  decoder_pipeline.decoder = torch.compile(decoder_pipeline.decoder, mode="reduce-overhead", fullgraph=True)
44
 
45
  if PREVIEW_IMAGES:
 
39
  decoder_pipeline.to(device)
40
 
41
  if USE_TORCH_COMPILE:
42
+ #prior_pipeline.prior = torch.compile(prior_pipeline.prior)
43
  decoder_pipeline.decoder = torch.compile(decoder_pipeline.decoder, mode="reduce-overhead", fullgraph=True)
44
 
45
  if PREVIEW_IMAGES: