Hugging Face's logo

Spaces:
fffiloni
/
zeroscope-XL
Paused

App Files Community
5
fffiloni commited on
Commit
741ef8a
·
1 Parent(s): a8a4d72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ pipe_xl = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_XL", torch_d
14
  pipe_xl.vae.enable_slicing()
15
  pipe_xl.scheduler = DPMSolverMultistepScheduler.from_config(pipe_xl.scheduler.config)
16
  pipe_xl.enable_model_cpu_offload()
17
- pipe_xl.to("cpu")
18
 
19
 
20
  def infer(prompt):
 
14
  pipe_xl.vae.enable_slicing()
15
  pipe_xl.scheduler = DPMSolverMultistepScheduler.from_config(pipe_xl.scheduler.config)
16
  pipe_xl.enable_model_cpu_offload()
17
+ pipe_xl.to("cuda")
18
 
19
 
20
  def infer(prompt):