Update Dockerfile

Dockerfile

@@ -1,5 +1,7 @@
+# Base image
 FROM python:3.10-slim
 
+# Non-interactive environment
 ENV DEBIAN_FRONTEND=noninteractive
 ENV VENV_PATH=/BitNet/venv
 ENV PATH="$VENV_PATH/bin:$PATH"
@@ -12,36 +14,41 @@ RUN apt-get update && apt-get install -y \
     cmake \
     build-essential \
     clang \
-    ca-certificates \
-    && rm -rf /var/lib/apt/lists/*
-RUN apt-get update && apt-get install -y \
-    wget curl git cmake build-essential clang libssl-dev libtinfo-dev \
+    libssl-dev \
+    libtinfo-dev \
     ca-certificates \
     && rm -rf /var/lib/apt/lists/*
 
-# Clone BitNet
-RUN git clone --recursive https://github.com/microsoft/BitNet.git
+# Clone BitNet repository
+RUN git clone --recursive https://github.com/microsoft/BitNet.git /BitNet
 
-# Copy local files
-COPY . /BitNet
+# Set working directory
 WORKDIR /BitNet
 
-# Create virtual environment
+# Copy local files (optional, if needed)
+COPY . /BitNet
+
+# Create Python virtual environment
 RUN python -m venv $VENV_PATH
 
-# Upgrade pip and install dependencies inside venv
+# Upgrade pip and install Python dependencies
 RUN pip install --upgrade pip
 RUN pip install -r requirements.txt
 RUN pip install huggingface_hub
 
+# Download model from Hugging Face
+RUN huggingface-cli download HF1BitLLM/Llama3-8B-1.58-100B-tokens \
+    --local-dir models/Llama3-8B-1.58-100B-tokens
 
-# Download model
-RUN huggingface-cli download HF1BitLLM/Llama3-8B-1.58-100B-tokens --local-dir models/Llama3-8B-1.58-100B-tokens
+# Setup environment / quantization in one go
+WORKDIR /BitNet/models/Llama3-8B-1.58-100B-tokens
+RUN python -m setup_env -md . -q i2_s
 
-# Setup environment / quantization
-RUN python -m setup_env -md models/Llama3-8B-1.58-100B-tokens -q i2_s
+# Optional: list files to confirm
+RUN ls -lah 
 
-RUN cd ./models/Llama3-8B-1.58-100B-tokens
-RUN ls
+# Reset working directory for app
+WORKDIR /BitNet
+RUN pwd
 # Default command
 CMD ["python", "app.py"]