Spaces:
Running
Running
:bug: bug fix use of cuda
Browse files
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):
|