Xenobd commited on
Commit
95bb6c2
·
verified ·
1 Parent(s): 460e404

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -39,10 +39,10 @@ ARG HF_TOKEN
39
  RUN huggingface-cli login --token $HF_TOKEN || true
40
 
41
  # Download the model into ./models/
42
- RUN huggingface-cli download microsoft/BitNet-b1.58-2B-4T-gguf --local-dir models/
43
  #RUN huggingface-cli download tiiuae/Falcon3-3B-Instruct-1.58bit-GGUF --local-dir models/
44
  #Xenobd/Bitnet-Test
45
- huggingface-cli download Xenobd/Bitnet-Test --local-dir models/
46
  # Run BitNet preprocessing script
47
  RUN python utils/codegen_tl1.py --model bitnet_b1_58-3B --BM 160,320,320 --BK 64,128,64 --bm 32,64,32
48
  RUN mkdir -p build && cd build && cmake .. && make -j$(nproc)
 
39
  RUN huggingface-cli login --token $HF_TOKEN || true
40
 
41
  # Download the model into ./models/
42
+ #RUN huggingface-cli download microsoft/BitNet-b1.58-2B-4T-gguf --local-dir models/
43
  #RUN huggingface-cli download tiiuae/Falcon3-3B-Instruct-1.58bit-GGUF --local-dir models/
44
  #Xenobd/Bitnet-Test
45
+ RUN huggingface-cli download Xenobd/Bitnet-Test --local-dir models/
46
  # Run BitNet preprocessing script
47
  RUN python utils/codegen_tl1.py --model bitnet_b1_58-3B --BM 160,320,320 --BK 64,128,64 --bm 32,64,32
48
  RUN mkdir -p build && cd build && cmake .. && make -j$(nproc)