Upload Dockerfile with huggingface_hub

Dockerfile

@@ -0,0 +1,27 @@
+FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
+
+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/*
+
+WORKDIR /app
+
+# Install dependencies
+RUN pip3 install --no-cache-dir \
+    torch>=2.4.0 \
+    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
+
+COPY train.py /app/train.py
+
+# Run training on startup
+CMD ["python3", "train.py"]