Spaces:
Runtime error
Runtime error
ehristoforu
commited on
Commit
β’
1a4173f
1
Parent(s):
9e7daa2
Update app.py
Browse files
app.py
CHANGED
@@ -84,7 +84,7 @@ if torch.cuda.is_available():
|
|
84 |
torch_dtype=torch.float16,
|
85 |
use_safetensors=True,
|
86 |
)
|
87 |
-
pipe_xl_lightning.scheduler = DPMSolverSinglestepScheduler.from_config(pipe_xl_lightning.scheduler.config, use_karras_sigmas=False, timestep_spacing="trailing")
|
88 |
pipe_xl_lightning.to(device)
|
89 |
|
90 |
print("Loaded on Device!")
|
|
|
84 |
torch_dtype=torch.float16,
|
85 |
use_safetensors=True,
|
86 |
)
|
87 |
+
pipe_xl_lightning.scheduler = DPMSolverSinglestepScheduler.from_config(pipe_xl_lightning.scheduler.config, use_karras_sigmas=False, timestep_spacing="trailing", lower_order_final=True)
|
88 |
pipe_xl_lightning.to(device)
|
89 |
|
90 |
print("Loaded on Device!")
|