Upload Dockerfile with huggingface_hub

Dockerfile

@@ -27,11 +27,11 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON
 # Upgrade pip
 RUN python -m pip install --no-cache-dir --upgrade pip setuptools wheel
 
-# Install PyTorch FIRST (required for flash-attn compilation)
+# Install PyTorch 2.5.1+ FIRST (required for xfuser's torch.distributed.tensor.experimental)
 RUN pip install --no-cache-dir \
-    torch==2.4.1 \
-    torchvision==0.19.1 \
-    torchaudio==2.4.1 \
+    torch==2.5.1 \
+    torchvision==0.20.1 \
+    torchaudio==2.5.1 \
     --index-url https://download.pytorch.org/whl/cu121
 
 # Try to install flash-attn, but don't fail if it doesn't work