yonigozlan HF staff commited on
Commit
c75d2db
·
1 Parent(s): a4bd236

change compile options

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ model = AutoModelForObjectDetection.from_pretrained(
22
  ).to(device)
23
  model_compiled = torch.compile(
24
  model,
25
- options={"triton.cudagraphs": True, "max_autotune": True},
26
  )
27
 
28
 
 
22
  ).to(device)
23
  model_compiled = torch.compile(
24
  model,
25
+ mode="reduce-overhead",
26
  )
27
 
28