ML-Project / requirements.txt
KenLumod's picture
Create requirements.txt
c064e39 verified
raw
history blame contribute delete
200 Bytes
# requirements.txt
torch>=2.0.1
transformers>=4.36.0
gradio>=4.0.0
accelerate>=0.25.0 # For optimized inference
sentencepiece>=0.2.0 # Required for tokenizers
protobuf>=3.20.0 # Required dependency