Spaces:
Running
on
Zero
Running
on
Zero
Update app_v3.py
Browse files
app_v3.py
CHANGED
@@ -50,7 +50,7 @@ pipe.to("cuda")
|
|
50 |
# For FLUX models, compiling VAE decode can also be beneficial if needed, though UNet is primary.
|
51 |
# pipe.vae.decode = torch.compile(pipe.vae.decode, mode="reduce-overhead", fullgraph=True) # Uncomment if VAE compile helps
|
52 |
try:
|
53 |
-
pipe.vae.decode = torch.compile(pipe.vae.decode, mode="
|
54 |
except Exception as e:
|
55 |
print(f"Compile failed: {e}")
|
56 |
|
|
|
50 |
# For FLUX models, compiling VAE decode can also be beneficial if needed, though UNet is primary.
|
51 |
# pipe.vae.decode = torch.compile(pipe.vae.decode, mode="reduce-overhead", fullgraph=True) # Uncomment if VAE compile helps
|
52 |
try:
|
53 |
+
pipe.vae.decode = torch.compile(pipe.vae.decode, dynamic=True, mode="default")
|
54 |
except Exception as e:
|
55 |
print(f"Compile failed: {e}")
|
56 |
|