multimodalart HF staff commited on
Commit
9b6c253
1 Parent(s): e16e0d9

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="max-autotune", 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, 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