word_keeper / requirements.txt
aaurelions's picture
Update requirements.txt
6ba0a9b verified
raw
history blame contribute delete
466 Bytes
torch>=2.1.0 --index-url https://download.pytorch.org/whl/cpu
transformers[sentencepiece]>=4.40.0 # Added sentencepiece here
accelerate>=0.29.0 # Or even try the latest version
# bitsandbytes # Optional for CPU, can remove if causing issues
peft>=0.10.0
# trl>=0.8.0 # Not strictly needed for inference app, only for training
gradio>=4.15.0 # Try a version known to be stable or newer if buttons are desired
numpy==1.26.4
protobuf # Often a dependency for tokenizers