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

add stable3 to models

add-new-stability-ai
Konstantine Tsafatinos 2024-09-23 11:56:58 -04:00
parent efb3bac811
commit b1aa9f697e
1 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
skynet/constants.py
View File

@ -17,9 +17,11 @@ MODELS = {
'Envvi/Inkpunk-Diffusion': {'short': 'ink', 'mem': 6},
'nousr/robo-diffusion': {'short': 'robot', 'mem': 6},
# Note: not sure about mem
'black-forest-labs/FLUX.1-schnell': {'short': 'flux', 'mem': 12},
'black-forest-labs/FLUX.1-dev': {'short': 'flux', 'mem': 8.3},
'stabilityai/stable-diffusion-3-medium-diffusers': {'short': 'stable3', 'mem': 8.3},
# default is always last
'stabilityai/stable-diffusion-xl-base-1.0': {'short': 'stablexl', 'mem': 8.3},
}
SHORT_NAMES = [