Spaces:
Sleeping
Sleeping
numpy force pip install in dockerfile
Browse files- Dockerfile +1 -0
- qwen_classifier/config.py +1 -1
- 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 =
|
|
|
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
|