Spaces:

nroggendorff
/

obama

Sleeping

nroggendorff commited on Apr 10

Commit

66cafa8

•

1 Parent(s): 138b6f5

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/obama", 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/obama", use_safetensors=True)
+    pipe = pipeline.to("cuda")
     image = pipe(num_inference_steps=steps).images[0]
     return image