59 lines
1.1 KiB
Python
59 lines
1.1 KiB
Python
from asyncio import PriorityQueue
|
|
from telethon import TelegramClient
|
|
import logging
|
|
import sqlite3
|
|
import coloredlogs
|
|
import asyncio
|
|
from process_queue import *
|
|
from config_priv import *
|
|
|
|
coloredlogs.install(level='INFO')
|
|
|
|
client = TelegramClient('bot', api_id, api_hash)
|
|
|
|
conn = sqlite3.connect('ai621.db', isolation_level=None)
|
|
conn.row_factory = sqlite3.Row
|
|
|
|
client.log_channel_id = log_channel_id
|
|
client.main_channel_id = main_channel_id
|
|
client.media_lock = asyncio.Lock()
|
|
|
|
client.admin_id = admin_id
|
|
|
|
log = logging.getLogger('bot')
|
|
|
|
send_prompt_lock = asyncio.Lock()
|
|
|
|
enabled_models = ['Yiffy', 'Furry Epoch', 'Zack3D']
|
|
|
|
fields_template = {
|
|
'prompt': None,
|
|
'sampler_name': 'k_euler_a',
|
|
'cfg_scale': 5.5,
|
|
'seed': 'random',
|
|
'resolution': '512x512',
|
|
'use_upscaling': False,
|
|
'steps': 50,
|
|
'n': 3,
|
|
'models': 'Yiffy',
|
|
'negative_prompt': None,
|
|
'image': None
|
|
}
|
|
|
|
prompt_template = {
|
|
'prompt': None, # ok
|
|
'params': {
|
|
'sampler_name': 'k_euler_a',
|
|
'toggles': [1, 4],
|
|
'cfg_scale': 5,
|
|
'seed': 'random',
|
|
'height': 512,
|
|
'width': 512,
|
|
'seed_variation': 1,
|
|
'use_upscaling': False,
|
|
'steps': 50,
|
|
'n': 3,
|
|
'models': 'Yiffy'
|
|
}
|
|
}
|