abhinit21 commited on
Commit
45df6df
1 Parent(s): 18df515

:bug: bug fix use of cuda

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -4,7 +4,6 @@ from diffusers import StableDiffusionLDM3DPipeline
4
 
5
  # Load the model. Do this once to avoid reloading on every request.
6
  pipe = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d-pano")
7
- pipe.to("cuda")
8
 
9
 
10
  def generate_images(prompt, guidance_scale=7.0, num_inference_steps=50):
 
4
 
5
  # Load the model. Do this once to avoid reloading on every request.
6
  pipe = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d-pano")
 
7
 
8
 
9
  def generate_images(prompt, guidance_scale=7.0, num_inference_steps=50):