LPX55 commited on
Commit
87ab7ec
·
verified ·
1 Parent(s): 993bc72

Update app_v3.py

Browse files
Files changed (1) hide show
  1. app_v3.py +1 -1
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="reduce-overhead", fullgraph=True)
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