File size: 3,549 Bytes
76dc483 900721f 76dc483 0f865da 0dae5e8 76dc483 0dae5e8 5d7e253 76dc483 5d7e253 14fe195 76dc483 4917b55 76dc483 4917b55 14fe195 eaf3496 0f865da 76dc483 900721f 76dc483 f108655 900721f f108655 76dc483 0dae5e8 76dc483 0dae5e8 76dc483 5d7e253 76dc483 0dae5e8 76dc483 0dae5e8 76dc483 0dae5e8 76dc483 0dae5e8 76dc483 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
import os
from constants import LOAD_DIFFUSERS_FORMAT_MODEL as LOAD_DIFFUSERS_FORMAT_MODEL_DC
CIVITAI_API_KEY = os.environ.get("CIVITAI_API_KEY")
HF_TOKEN = os.environ.get("HF_TOKEN")
HF_READ_TOKEN = os.environ.get('HF_READ_TOKEN') # only use for private repo
# - **List Models**
LOAD_DIFFUSERS_FORMAT_MODEL = [
'cagliostrolab/animagine-xl-3.1',
'Laxhar/noobai-XL-1.0',
'KBlueLeaf/Kohaku-XL-Epsilon-rev2',
'KBlueLeaf/Kohaku-XL-Epsilon-rev3',
'KBlueLeaf/Kohaku-XL-Zeta',
'kkayfahaarukku/UrangDiffusion-2.0',
'kayfahaarukku/irAsu-1.0',
'Eugeoter/artiwaifu-diffusion-2.0',
'comin/IterComp',
'OnomaAIResearch/Illustrious-xl-early-release-v0',
'Raelina/Rae-Diffusion-XL-V2',
'Raelina/Raemu-XL-V4',
'Raelina/Raehoshi-illust-XL',
'Raelina/Raehoshi-illust-xl-2',
'Raelina/Raehoshi-Illust-XL-2.1',
"camenduru/FLUX.1-dev-diffusers",
"black-forest-labs/FLUX.1-schnell",
"sayakpaul/FLUX.1-merged",
"ostris/OpenFLUX.1",
"multimodalart/FLUX.1-dev2pro-full",
"Raelina/Raemu-Flux",
]
LOAD_DIFFUSERS_FORMAT_MODEL = LOAD_DIFFUSERS_FORMAT_MODEL + LOAD_DIFFUSERS_FORMAT_MODEL_DC
DIFFUSERS_FORMAT_LORAS = [
"nerijs/animation2k-flux",
"XLabs-AI/flux-RealismLora",
"Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design",
]
# List all Models for specified user
HF_MODEL_USER_LIKES = ["votepurchase"] # sorted by number of likes
HF_MODEL_USER_EX = ["John6666"] # sorted by a special rule
# - **Download Models**
DOWNLOAD_MODEL_LIST = [
]
# - **Download VAEs**
DOWNLOAD_VAE_LIST = [
'https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/resolve/main/sdxl.vae.safetensors?download=true',
'https://huggingface.co/nubby/blessed-sdxl-vae-fp16-fix/resolve/main/sdxl_vae-fp16fix-c-1.1-b-0.5.safetensors?download=true',
'https://huggingface.co/nubby/blessed-sdxl-vae-fp16-fix/resolve/main/sdxl_vae-fp16fix-blessed.safetensors?download=true',
"https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt",
"https://huggingface.co/stabilityai/sd-vae-ft-ema-original/resolve/main/vae-ft-ema-560000-ema-pruned.ckpt",
]
# - **Download LoRAs**
DOWNLOAD_LORA_LIST = [
]
# Download Embeddings
DOWNLOAD_EMBEDS = [
'https://huggingface.co/datasets/Nerfgun3/bad_prompt/blob/main/bad_prompt_version2.pt',
'https://huggingface.co/embed/negative/resolve/main/EasyNegativeV2.safetensors',
'https://huggingface.co/embed/negative/resolve/main/bad-hands-5.pt',
]
DIRECTORY_MODELS = 'models'
DIRECTORY_LORAS = 'loras'
DIRECTORY_VAES = 'vaes'
DIRECTORY_EMBEDS = 'embedings'
DIRECTORY_EMBEDS_SDXL = 'embedings_xl'
DIRECTORY_EMBEDS_POSITIVE_SDXL = 'embedings_xl/positive'
directories = [DIRECTORY_MODELS, DIRECTORY_LORAS, DIRECTORY_VAES, DIRECTORY_EMBEDS, DIRECTORY_EMBEDS_SDXL, DIRECTORY_EMBEDS_POSITIVE_SDXL]
for directory in directories:
os.makedirs(directory, exist_ok=True)
HF_LORA_PRIVATE_REPOS1 = ['John6666/loratest1', 'John6666/loratest3', 'John6666/loratest4', 'John6666/loratest6']
HF_LORA_PRIVATE_REPOS2 = ['John6666/loratest10', 'John6666/loratest11','John6666/loratest'] # to be sorted as 1 repo
HF_LORA_PRIVATE_REPOS = HF_LORA_PRIVATE_REPOS1 + HF_LORA_PRIVATE_REPOS2
HF_LORA_ESSENTIAL_PRIVATE_REPO = 'John6666/loratest1' # to be downloaded on run app
HF_VAE_PRIVATE_REPO = 'John6666/vaetest'
HF_SDXL_EMBEDS_NEGATIVE_PRIVATE_REPO = 'John6666/embeddingstest'
HF_SDXL_EMBEDS_POSITIVE_PRIVATE_REPO = 'John6666/embeddingspositivetest'
|