Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
|
| 20 |
device_map='auto',
|
| 21 |
low_cpu_mem_usage=True,
|
| 22 |
offload_state_dict=True,
|
| 23 |
-
).to('cuda')
|
| 24 |
pipe.scheduler = EulerAncestralDiscreteScheduler(
|
| 25 |
beta_start=0.00085,
|
| 26 |
beta_end=0.012,
|
|
|
|
| 20 |
device_map='auto',
|
| 21 |
low_cpu_mem_usage=True,
|
| 22 |
offload_state_dict=True,
|
| 23 |
+
).to('cuda').to(torch.float16)
|
| 24 |
pipe.scheduler = EulerAncestralDiscreteScheduler(
|
| 25 |
beta_start=0.00085,
|
| 26 |
beta_end=0.012,
|