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
|