Spaces:
Running
Running
Commit
•
0e202ad
1
Parent(s):
2eadcf3
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ def load_model():
|
|
9 |
"sd-community/sdxl-flash",
|
10 |
torch_dtype=torch.float16,
|
11 |
low_cpu_mem_usage=True
|
12 |
-
)
|
13 |
pipe.scheduler = DPMSolverSinglestepScheduler.from_config(
|
14 |
pipe.scheduler.config,
|
15 |
timestep_spacing="trailing"
|
|
|
9 |
"sd-community/sdxl-flash",
|
10 |
torch_dtype=torch.float16,
|
11 |
low_cpu_mem_usage=True
|
12 |
+
).to(0)
|
13 |
pipe.scheduler = DPMSolverSinglestepScheduler.from_config(
|
14 |
pipe.scheduler.config,
|
15 |
timestep_spacing="trailing"
|