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

update constants and transformers version to work with stable 3

add-new-stability-ai
Konstantine Tsafatinos 2024-09-23 12:48:44 -04:00
parent b1aa9f697e
commit 8fe920a91d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@ -51,7 +51,7 @@ realesrgan = {version = '^0.3.0'}
quart-trio = {version = '^0.11.0'} quart-trio = {version = '^0.11.0'}
torchvision = {version = '0.15.2+cu118', source = 'torch'} torchvision = {version = '0.15.2+cu118', source = 'torch'}
accelerate = {version = '^0.32.0'} accelerate = {version = '^0.32.0'}
transformers = {version = '^4.33.2'} transformers = {version = '^4.37.2'}
sentencepiece = {version = '^0.2.0'} sentencepiece = {version = '^0.2.0'}
huggingface-hub = {version = '^0.24.5'} huggingface-hub = {version = '^0.24.5'}
invisible-watermark = {version = '^0.2.0'} invisible-watermark = {version = '^0.2.0'}

2
skynet/constants.py
View File

@ -17,7 +17,7 @@ MODELS = {
'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 # Note: not sure about mem
'black-forest-labs/FLUX.1-dev': {'short': 'flux', 'mem': 8.3}, # 'black-forest-labs/FLUX.1-dev': {'short': 'flux', 'mem': 8.3},
'stabilityai/stable-diffusion-3-medium-diffusers': {'short': 'stable3', 'mem': 8.3}, 'stabilityai/stable-diffusion-3-medium-diffusers': {'short': 'stable3', '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},