diff --git a/bot.py b/bot.py index 2ec641c..104bd12 100644 --- 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: diff --git a/config.json b/config.json index 7814332..6f4cc7c 100644 --- 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 } \ No newline at end of file