Upload Dockerfile with huggingface_hub

Dockerfile

@@ -1,20 +1,9 @@
-FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-devel
-
-ENV DEBIAN_FRONTEND=noninteractive
-
-RUN apt-get update && apt-get install -y git curl && rm -rf /var/lib/apt/lists/*
+FROM unsloth/unsloth:latest
 
 WORKDIR /app
 
-# 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
+# Additional deps
+RUN pip install --no-cache-dir einops jaxtyping
 
 COPY train.py /app/train.py