KeivanR commited on
Commit
9544219
·
1 Parent(s): c8a90a4

numpy force pip install in dockerfile

Browse files
Files changed (3) hide show
  1. Dockerfile +1 -0
  2. qwen_classifier/config.py +1 -1
  3. requirements.txt +3 -4
Dockerfile CHANGED
@@ -22,6 +22,7 @@ COPY app.py .
22
  COPY qwen_classifier/ ./qwen_classifier/
23
  COPY setup.py .
24
  RUN pip install --no-cache-dir -r requirements.txt
 
25
 
26
  # Install PyTorch with CUDA support
27
  RUN pip install --no-cache-dir \
 
22
  COPY qwen_classifier/ ./qwen_classifier/
23
  COPY setup.py .
24
  RUN pip install --no-cache-dir -r requirements.txt
25
+ RUN pip install --no-cache-dir numpy
26
 
27
  # Install PyTorch with CUDA support
28
  RUN pip install --no-cache-dir \
qwen_classifier/config.py CHANGED
@@ -19,4 +19,4 @@ TAG_NAMES = [
19
  'trees'
20
  ]
21
 
22
- EVAL_LIMIT = 20
 
19
  'trees'
20
  ]
21
 
22
+ EVAL_LIMIT = 8
requirements.txt CHANGED
@@ -1,5 +1,4 @@
1
- # Core requirements
2
- numpy>=1.23.0 # Explicitly added
3
  torch>=2.1.0
4
  transformers>=4.36.0
5
  click>=8.1.0
@@ -9,11 +8,11 @@ requests>=2.31.0
9
  pandas>=2.0.0
10
  pydantic>=2.0.0
11
  datasets>=2.14.0
12
- matplotlib>=3.7.0
13
 
14
  # Additional Space-specific requirements
15
  fastapi>=0.95.0
16
  uvicorn>=0.22.0
17
  accelerate>=0.24.0
18
  sentencepiece>=0.1.99
19
- auto-gptq>=0.5.0; sys_platform != 'win32'
 
1
+ # Core requirements from setup.py
 
2
  torch>=2.1.0
3
  transformers>=4.36.0
4
  click>=8.1.0
 
8
  pandas>=2.0.0
9
  pydantic>=2.0.0
10
  datasets>=2.14.0
11
+ matplotlib>=3.7.0 # Added visualization library
12
 
13
  # Additional Space-specific requirements
14
  fastapi>=0.95.0
15
  uvicorn>=0.22.0
16
  accelerate>=0.24.0
17
  sentencepiece>=0.1.99
18
+ auto-gptq>=0.5.0; sys_platform != 'win32' # Windows compatibility