Spaces:

sd-concepts-library
/

stable-diffusion-conceptualizer

Running on A10G

multimodalart HF Staff commited on Sep 15, 2022

Commit

c0391de

1 Parent(s): 3ad34f6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -147,10 +147,11 @@ def checkbox_block():
   return checkbox
 def infer(text):
-  images_list = pipe(
-      [text]*2,
-      num_inference_steps=50,
-      guidance_scale=7.5
   )
   output_images = []
   for i, image in enumerate(images_list["sample"]):

   return checkbox
 def infer(text):
+  with autocast("cuda"):
+    images_list = pipe(
+              [text]*2,
+              num_inference_steps=50,
+              guidance_scale=7.5
   )
   output_images = []
   for i, image in enumerate(images_list["sample"]):