File size: 4,764 Bytes
76dc483 900721f 76dc483 0f865da 0dae5e8 76dc483 0dae5e8 76dc483 14fe195 76dc483 4917b55 900721f 76dc483 4917b55 14fe195 eaf3496 0f865da 76dc483 900721f 76dc483 f108655 900721f f108655 76dc483 0dae5e8 76dc483 0dae5e8 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 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
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 = [
'votepurchase/animagine-xl-3.1',
'votepurchase/NSFW-GEN-ANIME-v2',
'votepurchase/kivotos-xl-2.0',
'votepurchase/holodayo-xl-2.1',
'votepurchase/ponyDiffusionV6XL',
'votepurchase/AnythingXL_xl',
'votepurchase/7thAnimeXLPonyA_v10',
'votepurchase/ChilloutMix',
'votepurchase/NovelAIRemix',
'votepurchase/NSFW-gen-v2',
'votepurchase/PerfectDeliberate-Anime_v2',
'votepurchase/realpony-xl',
'votepurchase/artiwaifu-diffusion-1.0',
'votepurchase/Starry-XL-v5.2',
'votepurchase/Yaki-Dofu-Mix',
'votepurchase/ebara-pony-v1-sdxl',
'votepurchase/waiANIMIXPONYXL_v10',
'votepurchase/counterfeitV30_v30',
'votepurchase/ebara-pony',
'votepurchase/Realistic_Vision_V1.4',
'votepurchase/pony',
'votepurchase/ponymatureSDXL_ponyeclipse10',
'votepurchase/waiREALMIX_v70',
'votepurchase/waiREALCN_v10',
'votepurchase/PVCStyleModelMovable_pony151',
'votepurchase/PVCStyleModelMovable_beta27Realistic',
'votepurchase/PVCStyleModelFantasy_beta12',
'votepurchase/pvcxl-v1-lora',
'votepurchase/Realistic_Vision_V2.0',
'votepurchase/RealVisXL_V4.0',
'votepurchase/juggernautXL_hyper_8step_sfw',
'votepurchase/ponyRealism_v21MainVAE',
'KBlueLeaf/Kohaku-XL-Epsilon-rev2',
'KBlueLeaf/Kohaku-XL-Epsilon-rev3',
'KBlueLeaf/Kohaku-XL-Zeta',
'kayfahaarukku/UrangDiffusion-2.0',
'kayfahaarukku/irAsu-1.0',
'Eugeoter/artiwaifu-diffusion-2.0',
'comin/IterComp',
'OnomaAIResearch/Illustrious-xl-early-release-v0',
'Laxhar/noobai-XL-1.0',
'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/blob/main/sdxl_vae-fp16fix-blessed.safetensors",
"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'
|