zeroscope-XL

Paused

fffiloni commited on Jul 3, 2023

Commit

741ef8a

1 Parent(s): a8a4d72

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ pipe_xl = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_XL", torch_d
 pipe_xl.vae.enable_slicing()
 pipe_xl.scheduler = DPMSolverMultistepScheduler.from_config(pipe_xl.scheduler.config)
 pipe_xl.enable_model_cpu_offload()
-pipe_xl.to("cpu")
 def infer(prompt):

 pipe_xl.vae.enable_slicing()
 pipe_xl.scheduler = DPMSolverMultistepScheduler.from_config(pipe_xl.scheduler.config)
 pipe_xl.enable_model_cpu_offload()
+pipe_xl.to("cuda")
 def infer(prompt):