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

add test txt2txt model to constants

txt2txt
Konstantine Tsafatinos 2025-01-09 21:32:02 -05:00
parent 76b0e99874
commit cc430360c1
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
skynet/constants.py
View File

@ -14,7 +14,7 @@ class Size(msgspec.Struct):
class ModelDesc(msgspec.Struct): class ModelDesc(msgspec.Struct):
short: str short: str
mem: float mem: float
size: Size size: Size | None
tags: list[Literal['txt2img', 'img2img', 'inpaint']] tags: list[Literal['txt2img', 'img2img', 'inpaint']]
MODELS: dict[str, ModelDesc] = { MODELS: dict[str, ModelDesc] = {
@ -78,6 +78,12 @@ MODELS: dict[str, ModelDesc] = {
size=Size(w=512, h=512), size=Size(w=512, h=512),
tags=['txt2img'] tags=['txt2img']
), ),
'microsoft/DialoGPT-small': ModelDesc(
short='dgpt-sm',
mem=6,
size=None,
tags=['txt2txt']
),
'diffusers/stable-diffusion-xl-1.0-inpainting-0.1': ModelDesc( 'diffusers/stable-diffusion-xl-1.0-inpainting-0.1': ModelDesc(
short='stablexl-inpainting', short='stablexl-inpainting',
mem=8.3, mem=8.3,