noddysnots commited on
Commit
3a3edda
Β·
verified Β·
1 Parent(s): 9965443

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,9 +1,9 @@
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
 
1
+ torch>=2.0.0
2
+ transformers==4.38.2
3
  numpy>=1.24.0
4
  pytest>=7.3.1
5
  requests
6
  sentence-transformers
7
  gradio
8
+ accelerate
9
+ flash-attn --no-build-isolation # Ensures torch is available first