修改默认的迭代次数为20，和webui保持一致

2023-02-06 03:30:40 +08:00 · 2023-02-06 03:30:40 +08:00 · 58f46ea51e
parent 612ff773b5
commit 58f46ea51e
2 changed files with 3 additions and 2 deletions
--- a/bot.py
+++ b/bot.py
@ -118,7 +118,7 @@ def app_draw():
            else:
                W = 768
                H = 768
-                ITS = 50
+                ITS = config_json.get('DefaultDiffutionIterations', 20)
                NUM_PIC = 1

    if W > 2500 or H > 2500:
--- a/config.json
+++ b/config.json
@ -1,5 +1,6 @@
 {
    "OpenAI-API-Key" : "",
    "GPT-Model" : "text-davinci-003",
-    "Diffusion-Model" : "stabilityai/stable-diffusion-2-1"
+    "Diffusion-Model" : "stabilityai/stable-diffusion-2-1",
+    "DefaultDiffutionIterations" : 20
 }