Add minimal requirements file for training server

This commit is contained in:
Rick McEwen
2026-01-06 15:17:34 -05:00
parent 55abb1217c
commit 304d743df8

26
requirements-training.txt Normal file
View File

@ -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