Spaces:
Runtime error
Runtime error
Upload env.py
Browse files
env.py
ADDED
|
@@ -0,0 +1,98 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
|
| 3 |
+
|
| 4 |
+
CIVITAI_API_KEY = os.environ.get("CIVITAI_API_KEY")
|
| 5 |
+
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 6 |
+
hf_read_token = os.environ.get('HF_READ_TOKEN') # only use for private repo
|
| 7 |
+
|
| 8 |
+
|
| 9 |
+
num_loras = 3
|
| 10 |
+
num_cns = 2
|
| 11 |
+
|
| 12 |
+
|
| 13 |
+
models = [
|
| 14 |
+
"camenduru/FLUX.1-dev-diffusers",
|
| 15 |
+
"black-forest-labs/FLUX.1-schnell",
|
| 16 |
+
"sayakpaul/FLUX.1-merged",
|
| 17 |
+
"ostris/OpenFLUX.1",
|
| 18 |
+
"multimodalart/FLUX.1-dev2pro-full",
|
| 19 |
+
"John6666/flux1-dev-minus-v1-fp8-flux",
|
| 20 |
+
"John6666/hyper-flux1-dev-fp8-flux",
|
| 21 |
+
"John6666/blue-pencil-flux1-v021-fp8-flux",
|
| 22 |
+
"Raelina/Raemu-Flux",
|
| 23 |
+
"John6666/raemu-flux-v10-fp8-flux",
|
| 24 |
+
"John6666/copycat-flux-test-fp8-v11-fp8-flux",
|
| 25 |
+
"John6666/wai-ani-flux-v10forfp8-fp8-flux",
|
| 26 |
+
"John6666/flux-dev8-anime-nsfw-fp8-flux",
|
| 27 |
+
"John6666/nepotism-fuxdevschnell-v3aio-fp8-flux",
|
| 28 |
+
"John6666/sumeshi-flux1s-v002e-fp8-flux",
|
| 29 |
+
"John6666/fca-style-v33-x10-8step-fp8-flux",
|
| 30 |
+
"John6666/lyh-anime-v10f1-fp8-flux",
|
| 31 |
+
"John6666/lyh-dalle-anime-v12dalle-fp8-flux",
|
| 32 |
+
"John6666/lyh-anime-flux-v2a1-fp8-flux",
|
| 33 |
+
"John6666/glimmerkin-flux-cute-v10-fp8-flux",
|
| 34 |
+
"John6666/niji-style-flux-devfp8-fp8-flux",
|
| 35 |
+
"John6666/niji56-style-v3-fp8-flux",
|
| 36 |
+
"John6666/xe-anime-flux-v04-fp8-flux",
|
| 37 |
+
"John6666/xe-figure-flux-01-fp8-flux",
|
| 38 |
+
"John6666/xe-pixel-flux-01-fp8-flux",
|
| 39 |
+
"John6666/xe-guoman-flux-02-fp8-flux",
|
| 40 |
+
"John6666/carnival-unchained-v10-fp8-flux",
|
| 41 |
+
"John6666/real-flux-10b-schnell-fp8-flux",
|
| 42 |
+
"John6666/fluxunchained-artfulnsfw-fut516xfp8e4m3fnv11-fp8-flux",
|
| 43 |
+
"John6666/fastflux-unchained-t5f16-fp8-flux",
|
| 44 |
+
"John6666/iniverse-mix-xl-sfwnsfw-fluxdfp16nsfwv11-fp8-flux",
|
| 45 |
+
"John6666/nsfw-master-flux-lora-merged-with-flux1-dev-fp16-v10-fp8-flux",
|
| 46 |
+
"John6666/the-araminta-flux1a1-fp8-flux",
|
| 47 |
+
"John6666/acorn-is-spinning-flux-v11-fp8-flux",
|
| 48 |
+
"John6666/stoiqo-afrodite-fluxxl-f1dalpha-fp8-flux",
|
| 49 |
+
"John6666/real-horny-pro-fp8-flux",
|
| 50 |
+
"John6666/centerfold-flux-v20fp8e5m2-fp8-flux",
|
| 51 |
+
"John6666/jib-mix-flux-v208stephyper-fp8-flux",
|
| 52 |
+
"John6666/sapianf-nude-men-women-for-flux-v20fp16-fp8-flux",
|
| 53 |
+
"John6666/flux-asian-realistic-v10-fp8-flux",
|
| 54 |
+
"John6666/fluxasiandoll-v10-fp8-flux",
|
| 55 |
+
"John6666/xe-asian-flux-01-fp8-flux",
|
| 56 |
+
"John6666/fluxescore-dev-v10fp16-fp8-flux",
|
| 57 |
+
# "",
|
| 58 |
+
]
|
| 59 |
+
|
| 60 |
+
model_trigger = {
|
| 61 |
+
"Raelina/Raemu-Flux": "anime",
|
| 62 |
+
"John6666/raemu-flux-v10-fp8-flux": "anime",
|
| 63 |
+
"John6666/fca-style-v33-x10-8step-fp8-flux": "fca_style",
|
| 64 |
+
}
|
| 65 |
+
|
| 66 |
+
# List all Models for specified user
|
| 67 |
+
HF_MODEL_USER_LIKES = [] # sorted by number of likes
|
| 68 |
+
HF_MODEL_USER_EX = [] # sorted by a special rule
|
| 69 |
+
|
| 70 |
+
|
| 71 |
+
|
| 72 |
+
# - **Download Models**
|
| 73 |
+
download_model_list = [
|
| 74 |
+
]
|
| 75 |
+
|
| 76 |
+
# - **Download VAEs**
|
| 77 |
+
download_vae_list = [
|
| 78 |
+
]
|
| 79 |
+
|
| 80 |
+
# - **Download LoRAs**
|
| 81 |
+
download_lora_list = [
|
| 82 |
+
]
|
| 83 |
+
|
| 84 |
+
DIFFUSERS_FORMAT_LORAS = []
|
| 85 |
+
|
| 86 |
+
directory_models = 'models'
|
| 87 |
+
os.makedirs(directory_models, exist_ok=True)
|
| 88 |
+
directory_loras = 'loras'
|
| 89 |
+
os.makedirs(directory_loras, exist_ok=True)
|
| 90 |
+
directory_vaes = 'vaes'
|
| 91 |
+
os.makedirs(directory_vaes, exist_ok=True)
|
| 92 |
+
|
| 93 |
+
|
| 94 |
+
HF_LORA_PRIVATE_REPOS1 = []
|
| 95 |
+
HF_LORA_PRIVATE_REPOS2 = [] # to be sorted as 1 repo
|
| 96 |
+
HF_LORA_PRIVATE_REPOS = HF_LORA_PRIVATE_REPOS1 + HF_LORA_PRIVATE_REPOS2
|
| 97 |
+
HF_LORA_ESSENTIAL_PRIVATE_REPO = '' # to be downloaded on run app
|
| 98 |
+
HF_VAE_PRIVATE_REPO = ''
|