env.py

import os


INIT_MODELS = [
    # ("meta-llama/Llama-4-Scout-17B-16E-Instruct", "together"),
    ("Qwen/Qwen3-32B", "nebius"),
    ("Qwen/Qwen2.5-72B-Instruct", "together"),
    ("deepseek-ai/DeepSeek-R1", "together"),
    ("deepseek-ai/DeepSeek-V3", "nebius"),
    ("deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", "novita"),   
    ("deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B", "novita"),   
    ("deepseek-ai/DeepSeek-R1-Distill-Llama-8B", "novita"),  
    ('Qwen/Qwen2.5-7B-Instruct','together'),
    ('Qwen/Qwen2.5-VL-3B-Instruct','together'),
    ('Qwen/Qwen2.5-1.5B-Instruct','nebius'),
    ('google/gemma-2-27b-it','nebius'),
    ('google/gemma-2-9b-it','nebius'),
    ('google/gemma-2-2b-it','nebius'),
]
MODELS = [m[0] for m in INIT_MODELS]
TASK = os.getenv("TASK")
# With storage
HF_TOKEN = os.getenv("HF_TOKEN")
ORG_NAME = os.getenv("ORG_NAME")