noddysnots commited on
Commit
8e593da
Β·
verified Β·
1 Parent(s): ead33f1

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
- transformers==4.38.2 # Ensure compatibility
2
- torch>=2.0.0
3
  numpy>=1.24.0
4
  pytest>=7.3.1
5
  requests
6
  sentence-transformers
7
  gradio
8
  accelerate # Required for `device_map="auto"`
9
- flash-attn # Fixes missing package issue
 
1
+ torch>=2.0.0 # Install torch first
2
+ transformers==4.38.2 # Ensure compatibility with DeepSeek-R1
3
  numpy>=1.24.0
4
  pytest>=7.3.1
5
  requests
6
  sentence-transformers
7
  gradio
8
  accelerate # Required for `device_map="auto"`
9
+ flash-attn # Install after torch to avoid setup issues