Xenobd commited on
Commit
9467a08
·
verified ·
1 Parent(s): 49489fb

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 python -m pip install --upgrade pip
29
- RUN python -m pip install -r requirements.txt
30
- RUN python -m pip install huggingface_hub
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"]