diff --git a/skynet/constants.py b/skynet/constants.py index c8fbbba..f368f29 100755 --- a/skynet/constants.py +++ b/skynet/constants.py @@ -5,21 +5,23 @@ VERSION = '0.1a12' DOCKER_RUNTIME_CUDA = 'skynet:runtime-cuda' MODELS = { - 'prompthero/openjourney': {'short': 'midj', 'mem': 6}, - 'runwayml/stable-diffusion-v1-5': {'short': 'stable', 'mem': 6}, - 'stabilityai/stable-diffusion-2-1-base': {'short': 'stable2', 'mem': 6}, - 'snowkidy/stable-diffusion-xl-base-0.9': {'short': 'stablexl0.9', 'mem': 8.3}, - 'Linaqruf/anything-v3.0': {'short': 'hdanime', 'mem': 6}, - 'hakurei/waifu-diffusion': {'short': 'waifu', 'mem': 6}, - 'nitrosocke/Ghibli-Diffusion': {'short': 'ghibli', 'mem': 6}, - 'dallinmackay/Van-Gogh-diffusion': {'short': 'van-gogh', 'mem': 6}, - 'lambdalabs/sd-pokemon-diffusers': {'short': 'pokemon', 'mem': 6}, - 'Envvi/Inkpunk-Diffusion': {'short': 'ink', 'mem': 6}, - 'nousr/robo-diffusion': {'short': 'robot', 'mem': 6}, + 'prompthero/openjourney': {'short': 'midj', 'mem': 6}, + 'runwayml/stable-diffusion-v1-5': {'short': 'stable', 'mem': 6}, + 'stabilityai/stable-diffusion-2-1-base': {'short': 'stable2', 'mem': 6}, + 'snowkidy/stable-diffusion-xl-base-0.9': {'short': 'stablexl0.9', 'mem': 8.3}, + 'Linaqruf/anything-v3.0': {'short': 'hdanime', 'mem': 6}, + 'hakurei/waifu-diffusion': {'short': 'waifu', 'mem': 6}, + 'nitrosocke/Ghibli-Diffusion': {'short': 'ghibli', 'mem': 6}, + 'dallinmackay/Van-Gogh-diffusion': {'short': 'van-gogh', 'mem': 6}, + 'lambdalabs/sd-pokemon-diffusers': {'short': 'pokemon', 'mem': 6}, + 'Envvi/Inkpunk-Diffusion': {'short': 'ink', 'mem': 6}, + 'nousr/robo-diffusion': {'short': 'robot', 'mem': 6}, # Note: not sure about mem - 'black-forest-labs/FLUX.1-schnell': {'short': 'flux', 'mem': 12}, + '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}, + 'stabilityai/stable-diffusion-xl-base-1.0': {'short': 'stablexl', 'mem': 8.3}, + } SHORT_NAMES = [