Meltemi / requirements.txt
TzepChris's picture
Upload requirements.txt
2050f81 verified
--extra-index-url https://download.pytorch.org/whl/cu118
numpy==1.26.4
torch==2.1.2+cu118 # ή νεότερη έκδοση συμβατή με cu118/cu121
torchvision==0.16.2+cu118
torchaudio==2.1.2+cu118
# triton # Το triton μπορεί να είναι χρήσιμο για ορισμένες βελτιστοποιήσεις, αλλά δεν είναι πάντα απαραίτητο.
# timm==0.9.12 # Δεν φαίνεται να χρησιμοποιείται άμεσα, μπορεί να αφαιρεθεί αν δεν είναι εξάρτηση άλλου πακέτου.
transformers>=4.38.0 # ΣΥΝΙΣΤΑΤΑΙ ΑΝΑΒΑΘΜΙΣΗ για νεότερα μοντέλα όπως το Meltemi
chromadb==0.4.24 # ή νεότερη σταθερή έκδοση
scikit-learn==1.3.2
tqdm
sentencepiece # Απαιτείται από πολλά tokenizers, συμπεριλαμβανομένου του Llama/Meltemi
joblib
gradio==4.20.0 # ή νεότερη έκδοση
unicodedata2
scipy
accelerate>=0.28.0 # ΣΥΝΙΣΤΑΤΑΙ ΑΝΑΒΑΘΜΙΣΗ, χρήσιμο για μεγάλα μοντέλα
bitsandbytes>=0.41.0 # ΠΡΟΣΘΕΣΤΕ ΑΥΤΟ αν θέλετε να χρησιμοποιήσετε 8-bit/4-bit quantization για εξοικονόμηση μνήμης
# einops # Μπορεί να απαιτείται από ορισμένες υλοποιήσεις μοντέλων στο transformers
# Σημείωση: Βεβαιωθείτε ότι οι εκδόσεις είναι συμβατές μεταξύ τους.
# Η αναβάθμιση των transformers και accelerate είναι καλή ιδέα.