YOFO_cost_and_speed_analysis / requirements.txt
RonniRodriguez's picture
Initial commit of YOFO Safety Evaluator
2b259aa
raw
history blame contribute delete
315 Bytes
# Core dependencies
torch>=2.0.0
transformers>=4.35.0
datasets>=2.14.0
accelerate>=0.24.0
peft>=0.7.0 # For LoRA
# Data processing
pandas>=2.0.0
numpy>=1.24.0
tqdm>=4.65.0
# Evaluation
scikit-learn>=1.3.0
matplotlib>=3.7.0
seaborn>=0.12.0
# Utilities
python-dotenv>=1.0.0
huggingface-hub>=0.19.0
gradio>=4.0.0