Add post-training folder

post-training/VLM-R1/Dockerfile

@@ -0,0 +1,42 @@
+# Use the specified base image
+FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel
+
+# Set working directory
+WORKDIR /workspace
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    wget \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install flash-attn using pre-built wheel
+RUN pip install --no-cache-dir \
+    "https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp311-cp311-linux_x86_64.whl" || \
+    pip install flash-attn --no-build-isolation
+
+# Install additional required packages
+RUN pip install \
+    wandb==0.18.3 \
+    tensorboardx \
+    qwen_vl_utils \
+    torchvision \
+    git+https://github.com/huggingface/transformers.git
+
+# Copy local open-r1-multimodal repository
+COPY ./src/open-r1-multimodal /workspace/src/open-r1-multimodal
+
+# Install open_r1
+WORKDIR /workspace/src/open-r1-multimodal
+RUN pip install -e ".[dev]"
+WORKDIR /workspace
+
+# Install vllm
+RUN pip install vllm==0.7.2
+
+# Set environment variables for better Python output
+ENV PYTHONUNBUFFERED=1
+
+# Default command
+CMD ["/bin/bash"]
+