| | import os |
| |
|
| | MODEL_NAME = os.getenv("MODEL_NAME", "openai/gpt-oss-120b:cerebras") |
| | SECONDARY_MODEL_NAME = os.getenv("SECONDARY_MODEL_NAME", "mistralai/Mixtral-8x7B-Instruct-v0.1") |
| | TERTIARY_MODEL_NAME = os.getenv("TERTIARY_MODEL_NAME", "llama/Llama-3.1-8B-Instruct:featherless-ai") |
| | CLIP_BASE_MODEL = os.getenv("CLIP_BASE_MODEL", "Salesforce/blip-image-captioning-large") |
| | CLIP_LARGE_MODEL = os.getenv("CLIP_LARGE_MODEL", "openai/clip-vit-large-patch14") |
| | ASR_MODEL = os.getenv("ASR_MODEL", "openai/whisper-large-v3") |
| | TTS_MODEL = os.getenv("TTS_MODEL", "facebook/mms-tts-ara") |
| | IMAGE_GEN_MODEL = os.getenv("IMAGE_GEN_MODEL", "Qwen/Qwen2.5-VL-7B-Instruct:novita") |
| | SECONDARY_IMAGE_GEN_MODEL = os.getenv("SECONDARY_IMAGE_GEN_MODEL", "black-forest-labs/FLUX.1-dev") |
| |
|
| | ROUTER_API_URL = os.getenv("ROUTER_API_URL", "https://router.huggingface.co") |
| | API_ENDPOINT = os.getenv("API_ENDPOINT", "https://router.huggingface.co/v1") |
| | FALLBACK_API_ENDPOINT = os.getenv("FALLBACK_API_ENDPOINT", "https://api-inference.huggingface.co/v1") |
| | IMAGE_INFERENCE_API = os.getenv("IMAGE_INFERENCE_API", "https://api-inference.huggingface.co/models") |
| |
|
| | MODEL_ALIASES = { |
| | "advanced": MODEL_NAME, |
| | "standard": SECONDARY_MODEL_NAME, |
| | "light": TERTIARY_MODEL_NAME, |
| | "image_base": CLIP_BASE_MODEL, |
| | "image_advanced": CLIP_LARGE_MODEL, |
| | "audio": ASR_MODEL, |
| | "tts": TTS_MODEL, |
| | "image_gen": IMAGE_GEN_MODEL, |
| | "secondary_image_gen": SECONDARY_IMAGE_GEN_MODEL |
| | } |
