requirements-finetuning.txt

# Project Sanctuary - Fine-Tuning Dependencies
# This file contains CUDA-enabled ML dependencies for model fine-tuning.
# These are NOT required for general project development or CI/CD.
# 
# Installation: pip install -r requirements-finetuning.txt
# Note: Requires CUDA 12.6+ and should be installed in a dedicated virtual environment.

# Use the PyTorch CUDA 12.6 wheel index for PyTorch packages.
--extra-index-url https://download.pytorch.org/whl/cu126
# Add NVIDIA's index to find correct CUDA-enabled binaries.
--extra-index-url https://pypi.nvidia.com

# --- CORE ML & FINE-TUNING STACK ---
# Pinned to PyTorch 2.9.0 to match xformers dependency.
torch==2.9.1+cu126
torchvision==0.24.1+cu126
torchaudio==2.9.1+cu126

# Pinned to a known-good, compatible set.
transformers>=4.53.0
peft==0.18.0
trl==0.26.2
bitsandbytes==0.49.0
triton>=2.3.0
datasets==4.4.2
accelerate==1.12.0
xformers
llama-cpp-python

# Keras compatibility layer
tf-keras==2.20.1

# Model conversion dependencies
sentencepiece
protobuf

# --- MODEL PUBLISHING & DEPLOYMENT ---
huggingface-hub
pyyaml