Big-Data-Lab
/
WeChatBot
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

This commit is contained in:
HfCloud 2023-02-06 03:30:40 +08:00
parent 612ff773b5
commit 58f46ea51e
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bot.py
View File

@ -118,7 +118,7 @@ def app_draw():
else: else:
W = 768 W = 768
H = 768 H = 768
ITS = 50 ITS = config_json.get('DefaultDiffutionIterations', 20)
NUM_PIC = 1 NUM_PIC = 1
if W > 2500 or H > 2500: if W > 2500 or H > 2500:

3
config.json
View File

@ -1,5 +1,6 @@
{ {
"OpenAI-API-Key" : "", "OpenAI-API-Key" : "",
"GPT-Model" : "text-davinci-003", "GPT-Model" : "text-davinci-003",
"Diffusion-Model" : "stabilityai/stable-diffusion-2-1" "Diffusion-Model" : "stabilityai/stable-diffusion-2-1",
"DefaultDiffutionIterations" : 20
} }