ehristoforu commited on
Commit
1a4173f
1 Parent(s): 9e7daa2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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!")