cocktailpeanut commited on
Commit
668eb9a
1 Parent(s): 5b94f47
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -72,6 +72,7 @@ guidance_scale = 5
72
 
73
  # LCM
74
  if os.environ.get("LCM"):
 
75
  num_inference_steps = 10
76
  guidance_scale = 0
77
 
@@ -79,6 +80,8 @@ if os.environ.get("LCM"):
79
  pipe.fuse_lora()
80
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
81
 
 
 
82
  if device == 'mps':
83
  pipe.to("mps", torch_dtype)
84
  pipe.enable_attention_slicing()
 
72
 
73
  # LCM
74
  if os.environ.get("LCM"):
75
+ print("LCM")
76
  num_inference_steps = 10
77
  guidance_scale = 0
78
 
 
80
  pipe.fuse_lora()
81
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
82
 
83
+ print(f"default: num_inference_steps={num_inference_steps}, guidance_scale={guidance_scale}")
84
+
85
  if device == 'mps':
86
  pipe.to("mps", torch_dtype)
87
  pipe.enable_attention_slicing()