Update Dockerfile

Dockerfile

@@ -1,6 +1,5 @@
 FROM ubuntu:22.04
 
-# Set environment variables to avoid some prompts
 ENV DEBIAN_FRONTEND=noninteractive
 
 # Update & install system dependencies
@@ -17,9 +16,15 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # Install LLVM
-# Clone the BitNet repository
+# NOTE: Uncomment this if you actually want to install LLVM.
+# RUN bash -c "$(wget -O - https://apt.llvm.org/llvm.sh)"
+
+# Clone BitNet
 RUN git clone --recursive https://github.com/microsoft/BitNet.git
 
+# Copy everything from the local directory (where Dockerfile is) into the BitNet directory in the container
+COPY . /BitNet
+
 WORKDIR /BitNet
 
 # Install Python dependencies
@@ -29,17 +34,18 @@ RUN pip install --upgrade pip && \
 # Hugging Face CLI setup
 RUN pip install huggingface_hub
 
-# Optional: Pass HF_TOKEN as build-arg or mount config manually
-# If model requires authentication, make sure to set HF_TOKEN in build
+# Optional: HF token
 ARG HF_TOKEN
 RUN huggingface-cli login --token $HF_TOKEN || true
 
-# Download model into models/ folder
+# Download the model into ./models/
 RUN huggingface-cli download tiiuae/Falcon3-7B-Instruct-1.58bit-GGUF --local-dir models/
+
+# Run BitNet preprocessing script
 RUN python utils/codegen_tl1.py --model bitnet_b1_58-3B --BM 160,320,320 --BK 64,128,64 --bm 32,64,32
 
-# Run cmake build process (assumes a CMakeLists.txt is in place)
+# Build the project
 RUN mkdir -p build && cd build && cmake .. && make -j$(nproc)
 
-# Set default command
+# Default command
 CMD ["python", "app.py"]