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

update models to include flux

add-flux-model
Konstantine Tsafatinos 2024-08-06 17:33:53 -04:00
parent 8a415b450f
commit 60be53ca6a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
skynet/constants.py
View File

@ -16,7 +16,8 @@ MODELS = {
'lambdalabs/sd-pokemon-diffusers': {'short': 'pokemon', 'mem': 6}, 'lambdalabs/sd-pokemon-diffusers': {'short': 'pokemon', 'mem': 6},
'Envvi/Inkpunk-Diffusion': {'short': 'ink', 'mem': 6}, 'Envvi/Inkpunk-Diffusion': {'short': 'ink', 'mem': 6},
'nousr/robo-diffusion': {'short': 'robot', 'mem': 6}, 'nousr/robo-diffusion': {'short': 'robot', 'mem': 6},
# Note: not sure about mem
'black-forest-labs/FLUX.1-dev': {'short': 'flux', 'mem': 8.3},
# default is always last # default is always last
'stabilityai/stable-diffusion-xl-base-1.0': {'short': 'stablexl', 'mem': 8.3}, 'stabilityai/stable-diffusion-xl-base-1.0': {'short': 'stablexl', 'mem': 8.3},
} }