requirements.txt

# NEUROX - Neural Energy Vampire Model Training Requirements
# ==========================================================

# Core ML libraries
torch>=2.0.0
transformers>=4.40.0
datasets>=2.18.0
accelerate>=0.28.0
bitsandbytes>=0.43.0

# Training libraries (for fine-tuning)
trl>=0.8.0
peft>=0.10.0
unsloth>=2024.3

# Hugging Face Hub (for uploading)
huggingface_hub>=0.22.0

# Utilities
sentencepiece>=0.2.0
protobuf>=4.25.0
safetensors>=0.4.0

# Optional: Experiment tracking
# wandb>=0.16.0

# Optional: GGUF export for llama.cpp
# llama-cpp-python>=0.2.0

# Optional: Gradio for demo interface
# gradio>=4.0.0