Spaces:

nroggendorff
/

nebulae

Sleeping

nroggendorff commited on Apr 10

Commit

30fe654

•

1 Parent(s): ca79767

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from diffusers import DiffusionPipeline
 @spaces.GPU
 def generate_image(steps):
     pipeline = DiffusionPipeline.from_pretrained("nroggendorff/nebulae", use_safetensors=True)
-    pipe = pipeline#.to("cuda")
     image = pipe(num_inference_steps=steps).images[0]
     return image

 @spaces.GPU
 def generate_image(steps):
     pipeline = DiffusionPipeline.from_pretrained("nroggendorff/nebulae", use_safetensors=True)
+    pipe = pipeline.to("cuda")
     image = pipe(num_inference_steps=steps).images[0]
     return image