# MiniMind (Mind2) Requirements

# Core
torch>=2.1.0
numpy>=1.24.0

# Training
transformers>=4.35.0
datasets>=2.14.0
accelerate>=0.24.0
wandb>=0.15.0

# Optimization & Export
onnx>=1.14.0
onnxruntime>=1.16.0

# Utilities
tqdm>=4.65.0
pyyaml>=6.0
jsonlines>=3.1.0

# Optional: Flash Attention (install separately)
# pip install flash-attn --no-build-isolation

# Optional: For INT4 quantization
# auto-gptq>=0.4.0
# autoawq>=0.1.0

# Development
pytest>=7.4.0
black>=23.0.0
isort>=5.12.0