test_MIIA / requirements.txt
AC-Angelo93's picture
Update requirements.txt
b4b14b0 verified
raw
history blame
337 Bytes
gradio
transformers>=4.34.0 # Or a more recent version
torch>=2.0.0
huggingface_hub>=0.17.0
sentencepiece # Often a dependency for tokenizers
accelerate # For device_map="auto" and better performance
# bitsandbytes # Only if you plan to implement 8-bit or 4-bit quantization (advanced)
# optimum # For ONNX, quantization, etc. (advanced)