Update Dockerfile

Dockerfile

@@ -36,10 +36,13 @@ WORKDIR ${HOME}/app
 # Copy your app code
 COPY --chown=user . ${HOME}/app
 
+# Install packaging first to ensure it's available for flash-attn
+RUN python3 -m pip install --user --no-cache-dir packaging==24.1
+
 # Install torch with specific version for compatibility
 RUN python3 -m pip install --user --no-cache-dir torch==2.4.1+cu121 torchvision==0.19.1+cu121 torchaudio==2.4.1+cu121 --index-url https://download.pytorch.org/whl/cu121
 
-# Install requirements (includes packaging and flash-attn)
+# Install requirements (includes flash-attn and other dependencies)
 RUN python3 -m pip install --user --no-cache-dir -r requirements.txt
 
 # Verify flash-attn installation