Spaces:
Yuanshi
/
Running on Zero

Yuanshi commited on
Commit
8c25deb
·
verified ·
1 Parent(s): 291cbbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,10 +56,10 @@ def infer(
56
  ):
57
  print("Using model:", model)
58
  if model == "2k":
59
- # pipe.vae.enable_tiling(False)
60
  pipe.set_adapters("2k")
61
  elif model == "4k":
62
- # pipe.vae.enable_tiling(True)
63
  pipe.set_adapters(f"4k_{flux_model}")
64
 
65
  if randomize_seed:
 
56
  ):
57
  print("Using model:", model)
58
  if model == "2k":
59
+ pipe.vae.enable_tiling(True)
60
  pipe.set_adapters("2k")
61
  elif model == "4k":
62
+ pipe.vae.enable_tiling(True)
63
  pipe.set_adapters(f"4k_{flux_model}")
64
 
65
  if randomize_seed: