mahoon-legal-ai / requirements.txt
hajimammad's picture
Update requirements.txt
ac3b632 verified
raw
history blame
551 Bytes
# Core ML stack
transformers==4.44.2
accelerate==0.34.2
datasets==2.21.0
evaluate==0.4.2
torch==2.3.1
sentencepiece
# RAG stack
chromadb==0.5.5
sentence-transformers==3.0.1
rank-bm25==0.2.2
langdetect==1.0.9
# LoRA / QLoRA (اختیاری برای آموزش سبک)
peft==0.11.1
trl==0.11.4
bitsandbytes==0.43.3 ; platform_system=="Linux" and platform_machine=="x86_64" and python_version < "3.13"
# UI
gradio==5.47.0
gradio-client==1.13.2
# Logging
wandb==0.17.9
# Utilities
scikit-learn>=1.4
packaging>=24.1
protobuf<4
click<8.1
httpx==0.27.2