requirements.txt

# Core dependencies
openenv-core[core] @ git+https://github.com/meta-pytorch/OpenEnv.git@v0.2.1
fastapi>=0.104.0
uvicorn>=0.24.0
httpx>=0.25.0
pydantic>=2.0.0,<2.12

# Training dependencies
# NOTE: PyTorch is NOT listed here — it must be installed separately
# with the correct CUDA version. See setup.sh or run:
#   pip install torch --index-url https://download.pytorch.org/whl/cu121
trl>=0.15.0
peft>=0.7.0
transformers>=4.40.0
datasets>=2.16.0

# Weights & Biases (optional but recommended)
wandb>=0.16.0

# HuggingFace Hub (for model push)
huggingface-hub>=0.20.0

# Plots and metrics
matplotlib>=3.8.0

# UI
gradio>=5.0.0