linoyts HF staff commited on
Commit
e7bc566
·
verified ·
1 Parent(s): 85ba10e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -484,7 +484,7 @@ if __name__ == "__main__":
484
  # scheduler=eulera_scheduler,
485
  )
486
  pipe_control.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_control.scheduler.config)
487
- pipe_control.cuda()
488
 
489
  # try:
490
  # pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
 
484
  # scheduler=eulera_scheduler,
485
  )
486
  pipe_control.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_control.scheduler.config)
487
+ pipe_control.to("cuda")
488
 
489
  # try:
490
  # pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg