Compare commits
2 Commits
55abb1217c
...
c086e8bbf7
| Author | SHA1 | Date | |
|---|---|---|---|
| c086e8bbf7 | |||
| 304d743df8 |
23
requirements-training.txt
Normal file
23
requirements-training.txt
Normal file
@ -0,0 +1,23 @@
|
||||
# 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, opencv-python) 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
|
||||
|
||||
# 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