Add minimal requirements file for training server
This commit is contained in:
26
requirements-training.txt
Normal file
26
requirements-training.txt
Normal 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
|
||||||
Reference in New Issue
Block a user