cocktailpeanut commited on
Commit
1256c38
·
1 Parent(s): ccb097b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ num_inference_steps = 30
72
  guidance_scale = 5
73
 
74
  # LCM
75
- if os.environ.get("LCM"):
76
  print("LCM")
77
  num_inference_steps = 10
78
  guidance_scale = 0
 
72
  guidance_scale = 5
73
 
74
  # LCM
75
+ if os.environ.get("MODE") == "LCM":
76
  print("LCM")
77
  num_inference_steps = 10
78
  guidance_scale = 0