KeivanR commited on
Commit
0dd09fd
·
1 Parent(s): 8524cf7

requitements

Browse files
Files changed (1) hide show
  1. requirements.txt +13 -4
requirements.txt CHANGED
@@ -1,8 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
  fastapi>=0.95.0
2
  uvicorn>=0.22.0
3
- transformers>=4.36.0
4
  accelerate>=0.24.0
5
- huggingface-hub>=0.19.0
6
- torch>=2.1.0
7
  sentencepiece>=0.1.99
8
- auto-gptq>=0.5.0 # If using quantized model
 
1
+ # Core requirements from setup.py
2
+ torch>=2.1.0
3
+ transformers>=4.36.0
4
+ click>=8.1.0
5
+ scikit-learn>=1.3.0
6
+ huggingface-hub>=0.19.0
7
+ requests>=2.31.0
8
+ pandas>=2.0.0
9
+ pydantic>=2.0.0
10
+ datasets>=2.14.0
11
+
12
+ # Additional Space-specific requirements
13
  fastapi>=0.95.0
14
  uvicorn>=0.22.0
 
15
  accelerate>=0.24.0
 
 
16
  sentencepiece>=0.1.99
17
+ auto-gptq>=0.5.0; sys_platform != 'win32' # Windows compatibility