Spaces:
Running
on
A10G
Running
on
A10G
rynmurdock
commited on
Commit
•
f47c034
1
Parent(s):
97656de
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ start_time = time.time()
|
|
32 |
####################### Setup Model
|
33 |
model_id = "stabilityai/stable-diffusion-xl-base-1.0"
|
34 |
lcm_lora_id = "latent-consistency/lcm-lora-sdxl"
|
35 |
-
pipe = SDEmb.from_pretrained(model_id, variant="fp16", low_mem_cpu_usage=True
|
36 |
pipe.load_lora_weights(lcm_lora_id)
|
37 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
38 |
pipe.to(device='cuda', dtype=torch.float16)
|
|
|
32 |
####################### Setup Model
|
33 |
model_id = "stabilityai/stable-diffusion-xl-base-1.0"
|
34 |
lcm_lora_id = "latent-consistency/lcm-lora-sdxl"
|
35 |
+
pipe = SDEmb.from_pretrained(model_id, variant="fp16", low_mem_cpu_usage=True)
|
36 |
pipe.load_lora_weights(lcm_lora_id)
|
37 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
38 |
pipe.to(device='cuda', dtype=torch.float16)
|