kimhyunwoo commited on
Commit
b5f6ba9
·
verified ·
1 Parent(s): da1470a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -5
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
- transformers>=4.38.1 # Specify a minimum version that supports Gemma and Flash Attention 2
2
- gradio>=4.0.0 # Use a recent version of Gradio for gr.ChatInterface
3
- torch>=2.1.0 # Use a recent version of PyTorch (2.1+ is recommended)
4
- accelerate>=0.25.0 # For efficient model loading and device management
5
- safetensors # For loading .safetensors model files
 
 
1
+ transformers>=4.38.1
2
+ gradio>=4.0.0
3
+ torch>=2.1.0
4
+ accelerate>=0.25.0
5
+ safetensors
6
+ huggingface-hub>=0.20.0