diff --git a/requirements-training.txt b/requirements-training.txt new file mode 100644 index 0000000..3d4119e --- /dev/null +++ b/requirements-training.txt @@ -0,0 +1,26 @@ +# Requirements for CLIP logo fine-tuning on RTX 4090 +# +# Only includes packages not already installed on the training server. +# Does NOT upgrade existing packages (torch, torchvision, numpy, pillow, pyyaml) +# which are already installed and compatible. +# +# Usage: +# pip install -r requirements-training.txt + +# CLIP models and tokenizers +transformers>=4.36.0 + +# LoRA fine-tuning +peft>=0.7.0 + +# Progress bars +tqdm>=4.66.0 + +# Image processing for detection pipeline +opencv-python>=4.8.0 + +# HuggingFace Hub for model downloads +huggingface-hub>=0.19.0 + +# Accelerate for efficient training (optional but recommended) +accelerate>=0.25.0