Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
Commit
•
668eb9a
1
Parent(s):
5b94f47
update
Browse files
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()
|