multimodalart HF staff commited on
Commit
03fb31d
1 Parent(s): 696abc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,8 +39,8 @@ 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, mode="reduce-overhead", fullgraph=True)
43
- decoder_pipeline.decoder = torch.compile(decoder_pipeline.decoder, mode="max-autotune", fullgraph=True)
44
 
45
  if PREVIEW_IMAGES:
46
  pass
 
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:
46
  pass