Update Dockerfile

Dockerfile

@@ -14,12 +14,15 @@ WORKDIR $HOME/app
 # Copy the requirements file
 COPY --chown=user:user ./requirements.txt .
 
-# --- BUILD FIX ---
-# 1. Upgrade pip itself and install the key build-time dependency first.
+# --- BUILD FIX v3 ---
+# 1. Upgrade pip itself and install the key build-time dependencies first.
+#    flash-attn needs 'packaging' AND 'torch' to build itself.
 RUN pip install --no-cache-dir --upgrade pip setuptools
 RUN pip install --no-cache-dir packaging
+RUN pip install --no-cache-dir torch==2.1.2
 
-# 2. Now, install all the other requirements. pip will now find 'packaging'.
+# 2. Now, install all the other requirements.
+#    pip will use the already-installed torch and not try to reinstall it.
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 # Copy the rest of the application