Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +3 -8
Dockerfile
CHANGED
|
@@ -25,13 +25,10 @@ WORKDIR /BitNet
|
|
| 25 |
RUN python -m venv $VENV_PATH
|
| 26 |
|
| 27 |
# Upgrade pip and install dependencies inside venv
|
| 28 |
-
RUN
|
| 29 |
-
RUN
|
| 30 |
-
RUN
|
| 31 |
|
| 32 |
-
# Optional: Hugging Face token
|
| 33 |
-
ARG HF_TOKEN
|
| 34 |
-
RUN huggingface-cli login --token $HF_TOKEN || true
|
| 35 |
|
| 36 |
# Download model
|
| 37 |
RUN python -m huggingface-cli download HF1BitLLM/Llama3-8B-1.58-100B-tokens --local-dir models/Llama3-8B-1.58-100B-tokens
|
|
@@ -39,8 +36,6 @@ RUN python -m huggingface-cli download HF1BitLLM/Llama3-8B-1.58-100B-tokens --lo
|
|
| 39 |
# Setup environment / quantization
|
| 40 |
RUN python -m setup_env -md models/Llama3-8B-1.58-100B-tokens -q i2_s
|
| 41 |
|
| 42 |
-
# Build C++ runtime
|
| 43 |
-
RUN mkdir -p build && cd build && cmake .. && make -j$(nproc)
|
| 44 |
|
| 45 |
# Default command
|
| 46 |
CMD ["python", "app.py"]
|
|
|
|
| 25 |
RUN python -m venv $VENV_PATH
|
| 26 |
|
| 27 |
# Upgrade pip and install dependencies inside venv
|
| 28 |
+
RUN pip install --upgrade pip
|
| 29 |
+
RUN pip install -r requirements.txt
|
| 30 |
+
RUN pip install huggingface_hub
|
| 31 |
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
# Download model
|
| 34 |
RUN python -m huggingface-cli download HF1BitLLM/Llama3-8B-1.58-100B-tokens --local-dir models/Llama3-8B-1.58-100B-tokens
|
|
|
|
| 36 |
# Setup environment / quantization
|
| 37 |
RUN python -m setup_env -md models/Llama3-8B-1.58-100B-tokens -q i2_s
|
| 38 |
|
|
|
|
|
|
|
| 39 |
|
| 40 |
# Default command
|
| 41 |
CMD ["python", "app.py"]
|