Spaces:

filipvk
/

tiny-sd-fastapi

Sleeping

filipvk commited on Jun 11

Commit

db9ea67

1 Parent(s): a2776d1

change output size

Files changed (1) hide show

app/model.py CHANGED Viewed

@@ -19,6 +19,6 @@ pipe.to("cpu")
 @torch.inference_mode()
 def generate_image(prompt):
-    image = pipe(prompt, height=256, width=256, num_inference_steps=10).images[0]
     return image

 @torch.inference_mode()
 def generate_image(prompt):
+    image = pipe(prompt, height=512, width=512, num_inference_steps=10).images[0]
     return image