Update app.py
Browse files
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="
|
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(
|