# Core dependencies (from pyproject.toml)
torch==2.6.0
transformers==4.52.4

# Gradio web interface
gradio==5.9.1

# HuggingFace Hub for checkpoint downloads
huggingface-hub>=0.26.0

# Configuration file parsing
pyyaml>=6.0

# Additional dependencies for rosetta package
datasets>=2.0
accelerate>=0.20

# Utilities
einops>=0.8