Canstralian commited on
Commit
d73a5d2
1 Parent(s): f7e9cda

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  gradio==3.41.1
2
  huggingface-hub==0.22.2
3
  transformers==4.34.0
4
- torch>=2.0.0 # Required for Transformers (install based on your hardware; GPU support recommended)
5
- sentencepiece # Required for some Hugging Face models like T5, Bloom, or GPT models
 
1
  gradio==3.41.1
2
  huggingface-hub==0.22.2
3
  transformers==4.34.0
4
+ torch>=2.0.0 # Adjust version based on CPU/GPU compatibility
5
+ sentencepiece # Required for specific models like Bloom