| """Environment-derived configuration and global constants. | |
| Importing this module has no side effects beyond reading env vars and computing | |
| a handful of constants. It must not import torch, diffusers, or gradio so it can | |
| be imported anywhere without dragging in heavy dependencies. | |
| """ | |
| import os | |
| import numpy as np | |
| # ── Model variant selection ────────────────────────────────────────────────── | |
| MODEL_VARIANT = os.environ.get("MODEL_VARIANT", "9B") | |
| if MODEL_VARIANT == "9B-KV": | |
| _MODEL_REPO = "black-forest-labs/FLUX.2-klein-9b-kv" | |
| else: | |
| MODEL_VARIANT = "9B" # normalise any typos back to default | |
| _MODEL_REPO = "black-forest-labs/FLUX.2-klein-9B" | |
| MODEL_REPO = _MODEL_REPO | |
| # ── Misc constants ─────────────────────────────────────────────────────────── | |
| MAX_SEED = int(np.iinfo(np.int32).max) | |
| # Maximum number of simultaneous LoRA weight sliders to pre-render in the UI. | |
| MAX_LORA_SLOTS = 6 | |
| # External LoRA catalog (bucket-mounted JSON). Single source of truth for the | |
| # selectable list. If missing/empty, the app seeds it from built-in defaults. | |
| PERSISTENT_LORA_CATALOG_PATH = os.environ.get( | |
| "PERSISTENT_LORA_CATALOG_PATH", | |
| "/loras-flux/config/lorasplayground.json", | |
| ) | |
| # Abliterated (uncensored) text encoder — replaces the stock Qwen3 text encoder | |
| # weights after pipeline load. Downloaded via HF Hub for speed + caching. | |
| UNCENSORED_TE_REPO = "ponpoke/flux2-klein-9b-uncensored-text-encoder" | |
| UNCENSORED_TE_FILE = "model.safetensors" | |