Spaces:

John6666
/

text2tag-llm

Running on Zero

John6666 commited on Jul 28, 2024

Commit

7764a67

verified ·

1 Parent(s): 44dfbbd

Upload genimage.py

Files changed (1) hide show

genimage.py CHANGED Viewed

@@ -38,7 +38,7 @@ def generate_image(prompt, neg_prompt):
         "negative_prompt": neg_prompt,
         "resolution": f"{1024} x {1024}",
         "guidance_scale": 7.5,
-        "num_inference_steps": 16,
         "sampler": "Euler",
     }
     try:
@@ -48,7 +48,7 @@ def generate_image(prompt, neg_prompt):
             width=1024,
             height=1024,
             guidance_scale=7.5,
-            num_inference_steps=16,
             output_type="pil",
             clip_skip=1,
         ).images

         "negative_prompt": neg_prompt,
         "resolution": f"{1024} x {1024}",
         "guidance_scale": 7.5,
+        "num_inference_steps": 28,
         "sampler": "Euler",
     }
     try:
             width=1024,
             height=1024,
             guidance_scale=7.5,
+            num_inference_steps=28,
             output_type="pil",
             clip_skip=1,
         ).images