diff --git a/pyproject.toml b/pyproject.toml index 9e452f5..514391e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -51,7 +51,7 @@ realesrgan = {version = '^0.3.0'} quart-trio = {version = '^0.11.0'} torchvision = {version = '0.15.2+cu118', source = 'torch'} accelerate = {version = '^0.32.0'} -transformers = {version = '^4.33.2'} +transformers = {version = '^4.37.2'} sentencepiece = {version = '^0.2.0'} huggingface-hub = {version = '^0.24.5'} invisible-watermark = {version = '^0.2.0'} diff --git a/skynet/constants.py b/skynet/constants.py index f368f29..9e1755b 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-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},