blood-cell-classifier / requirements.txt
laurru01's picture
Upload requirements.txt with huggingface_hub
0534c7d verified
raw
history blame contribute delete
436 Bytes
# Base libraries
pip==23.1.2
torch==2.5.1
transformers==4.46.2
gradio>=4.19.2
Pillow==9.4.0
# Model and optimization related
peft
accelerate
bitsandbytes
triton
xformers==0.0.28.post3
trl
# Data processing and utilities
tqdm
python-dotenv
datasets
wandb
huggingface_hub
# Unsloth specific
unsloth @ git+https://github.com/unslothai/unsloth.git
unsloth_zoo
# Deep Learning frameworks
tf-keras
# Additional dependencies
typing
numpy