skygpu
/
skynet
mirror of https://github.com/skygpu/skynet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix default model and increments for discord ui

pull/19/head
Konstantine Tsafatinos 2023-08-23 11:53:41 -04:00
parent 35f8276e4e
commit 75268decc4
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
skynet/constants.py
View File

@ -158,7 +158,7 @@ DEFAULT_GUIDANCE = 7.5
DEFAULT_STRENGTH = 0.5 DEFAULT_STRENGTH = 0.5
DEFAULT_STEP = 28 DEFAULT_STEP = 28
DEFAULT_CREDITS = 10 DEFAULT_CREDITS = 10
DEFAULT_MODEL = 'stablexl' DEFAULT_MODEL = list(MODELS.keys())[4]
DEFAULT_ROLE = 'pleb' DEFAULT_ROLE = 'pleb'
DEFAULT_UPSCALER = None DEFAULT_UPSCALER = None

12
skynet/frontend/discord/ui.py
View File

@ -63,9 +63,9 @@ class Txt2ImgButton(discord.ui.Button):
await db_call( await db_call(
'update_user_stats', user.id, 'txt2img', last_prompt=prompt) 'update_user_stats', user.id, 'txt2img', last_prompt=prompt)
ec = await work_request(user, status_msg, 'txt2img', params, msg) success = await work_request(user, status_msg, 'txt2img', params, msg)
if ec == None: if success:
await db_call('increment_generated', user.id) await db_call('increment_generated', user.id)
@ -145,13 +145,13 @@ class Img2ImgButton(discord.ui.Button):
last_binary=ipfs_hash last_binary=ipfs_hash
) )
ec = await work_request( success = await work_request(
user, status_msg, 'img2img', params, msg, user, status_msg, 'img2img', params, msg,
file_id=file_id, file_id=file_id,
binary_data=ipfs_hash binary_data=ipfs_hash
) )
if ec == None: if success:
await db_call('increment_generated', user.id) await db_call('increment_generated', user.id)
@ -195,13 +195,13 @@ class RedoButton(discord.ui.Button):
'prompt': prompt, 'prompt': prompt,
**user_config **user_config
} }
ec = await work_request( success = await work_request(
user, status_msg, 'redo', params, interaction, user, status_msg, 'redo', params, interaction,
file_id=file_id, file_id=file_id,
binary_data=binary binary_data=binary
) )
if ec == None: if success:
await db_call('increment_generated', user.id) await db_call('increment_generated', user.id)