Hugging Face's logo

Spaces:
soiz1
/
Zero123PlusDemo
Runtime error

App Files Community
soiz1 commited on
Commit
3659cf2
·
verified ·
1 Parent(s): e4bd272

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ pipeline = DiffusionPipeline.from_pretrained(
22
  pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
23
  pipeline.scheduler.config, timestep_spacing='trailing'
24
  )
25
- pipeline.to('cuda:0')
26
 
27
 
28
  def inference(input_img, num_inference_steps, guidance_scale, seed ):
 
22
  pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
23
  pipeline.scheduler.config, timestep_spacing='trailing'
24
  )
25
+ pipeline.to('cpu')
26
 
27
 
28
  def inference(input_img, num_inference_steps, guidance_scale, seed ):