tgbots/TeleDiffusionBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Setting generation properties

Browse Source
This commit is contained in:
BarsTiger
2023-02-26 22:16:05 +02:00
parent 8f0cd0ac05
commit 3c3d644137
10 changed files with 286 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
bot/handlers/txt2img/__init__.py
View File

@@ -1,6 +1,19 @@
from bot.common import dp
from .txt2img import txt2img_comand
from .txt2img import generate_command
from .set_settings import (
set_height_command, set_negative_prompt_command, set_size_command, set_steps_command, set_width_command,
set_prompt_command, set_sampler_command, set_cfg_scale_command, set_restore_faces_command
)
def register():
dp.register_message_handler(txt2img.txt2img_comand, commands='txt2img')
dp.register_message_handler(txt2img.generate_command, commands='generate')
dp.register_message_handler(set_settings.set_prompt_command, commands='setprompt')
dp.register_message_handler(set_settings.set_height_command, commands='setheight')
dp.register_message_handler(set_settings.set_width_command, commands='setwidth')
dp.register_message_handler(set_settings.set_negative_prompt_command, commands='setnegative')
dp.register_message_handler(set_settings.set_size_command, commands='setsize')
dp.register_message_handler(set_settings.set_steps_command, commands='setsteps')
dp.register_message_handler(set_settings.set_sampler_command, commands='setsampler')
dp.register_message_handler(set_settings.set_cfg_scale_command, commands='setscale')
dp.register_message_handler(set_settings.set_restore_faces_command, commands='setfaces')