From 3e319829a7eb48008faea757906b655b7b8fa18b Mon Sep 17 00:00:00 2001 From: Konstantine Tsafatinos Date: Wed, 7 Aug 2024 18:06:52 -0400 Subject: [PATCH] update libs for flux, add flux to models --- pyproject.toml | 10 +++++++--- skynet/constants.py | 2 +- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 9a33d8f..1cb3d82 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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' diff --git a/skynet/constants.py b/skynet/constants.py index d003d95..c8fbbba 100755 --- a/skynet/constants.py +++ b/skynet/constants.py @@ -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}, }