Upload Dockerfile with huggingface_hub

Dockerfile

@@ -1,27 +1,21 @@
-FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
+FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-devel
 
 ENV DEBIAN_FRONTEND=noninteractive
-# HF_TOKEN will be set via Space Secrets
 
-RUN apt-get update && apt-get install -y \
-    python3 python3-pip git curl \
-    && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y git curl && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# Install dependencies
-RUN pip3 install --no-cache-dir \
-    torch>=2.4.0 \
+# Install dependencies (no flash-attn, use sdpa instead)
+RUN pip install --no-cache-dir \
     transformers>=4.46.0 \
     datasets>=3.0.0 \
     peft>=0.13.0 \
     trl>=0.12.0 \
     bitsandbytes>=0.44.0 \
     accelerate>=1.0.0 \
-    huggingface_hub>=0.26.0 \
-    flash-attn --no-build-isolation
+    huggingface_hub>=0.26.0
 
 COPY train.py /app/train.py
 
-# Run training on startup
-CMD ["python3", "train.py"]
+CMD ["python", "train.py"]