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

update libs for flux, add flux to models

add-flux-model
Konstantine Tsafatinos 2024-08-07 18:06:52 -04:00
parent c26871536a
commit 3e319829a7
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyproject.toml
View File

@ -45,13 +45,16 @@ triton = {version = '2.0.0', source = 'torch'}
basicsr = {version = '^1.4.2'}
xformers = {version = '^0.0.22'}
hypercorn = {version = '^0.14.4'}
diffusers = {version = '^0.21.2'}
# diffusers = {version = '^0.30.0.dev0'}
diffusers = {git = 'https://github.com/iamzoltan/diffusers.git', branch = 'flux-pipeline-vae-slicing-tiling'}
requests = {version = '2.31.0'}
realesrgan = {version = '^0.3.0'}
quart-trio = {version = '^0.11.0'}
torchvision = {version = '0.15.2+cu118', source = 'torch'}
accelerate = {version = '^0.23.0'}
accelerate = {version = '^0.32.0'}
transformers = {version = '^4.33.2'}
huggingface-hub = {version = '^0.17.3'}
sentencepiece = {version = '^0.2.0'}
huggingface-hub = {version = '^0.24.5'}
invisible-watermark = {version = '^0.2.0'}
[[tool.poetry.source]]
@ -65,3 +68,4 @@ build-backend = 'poetry.core.masonry.api'
[tool.poetry.scripts]
skynet = 'skynet.cli:skynet'
txt2img = 'skynet.cli:txt2img'

2
skynet/constants.py
View File

@ -17,7 +17,7 @@ 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-dev': {'short': 'flux', 'mem': 8.3},
'black-forest-labs/FLUX.1-schnell': {'short': 'flux', 'mem': 12},
# default is always last
'stabilityai/stable-diffusion-xl-base-1.0': {'short': 'stablexl', 'mem': 8.3},
}