Spaces:
Runtime error
Runtime error
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()
|