Eyalgut commited on
Commit
7faade7
1 Parent(s): 37f607f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ pipe.force_zeros_for_empty_prompt = False
29
  print("Optimizing BRIA-2.2 - this could take a while")
30
  t=time.time()
31
  pipe.unet = torch.compile(
32
- pipe.unet, mode="max-autotune", fullgraph=True # 600 secs compilation
33
  )
34
  with torch.no_grad():
35
  outputs = pipe(
 
29
  print("Optimizing BRIA-2.2 - this could take a while")
30
  t=time.time()
31
  pipe.unet = torch.compile(
32
+ pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
33
  )
34
  with torch.no_grad():
35
  outputs = pipe(