Update Dockerfile

Dockerfile

@@ -1,25 +1,32 @@
 # Unsloth Training Hub Dockerfile
 # CRITICAL: Installation order matters for Unsloth!
 
+
 FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
 
+
 # Prevent interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
 
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y     python3.11     python3.11-venv     python3-pip     git     wget     curl     && rm -rf /var/lib/apt/lists/*
 
+
 # Set Python 3.11 as default
 RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1
 RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1
 
+
 # Create virtual environment
 RUN python3 -m venv /opt/venv
 ENV PATH="/opt/venv/bin:$PATH"
 
+
 # Upgrade pip and install uv for faster installs
 RUN pip install --upgrade pip && pip install uv
 
+
 # ============================================================================
 # CRITICAL: UNSLOTH INSTALLATION ORDER
 # 1. Install PyTorch first
@@ -29,40 +36,52 @@ RUN pip install --upgrade pip && pip install uv
 # 5. Install TRL from git (latest)
 # ============================================================================
 
+
 # Step 1: PyTorch with CUDA 12.1
 RUN uv pip install torch==2.4.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
 
+
 # Step 2: Unsloth (main package)
-RUN uv pip install "unsloth[cu121-ampere-torch240] @ git+https://github.com/unslothai/unsloth.git"
+RUN pip install "unsloth[cu121-ampere-torch240] @ git+https://github.com/unslothai/unsloth.git"
+
 
 # Step 3: vLLM for fast inference during RL
 RUN uv pip install vllm
 
+
 # Step 4: diffusers (required for GRPO)
 RUN uv pip install diffusers
 
+
 # Step 5: TRL from git (latest for GRPO/GSPO support)
 RUN uv pip install git+https://github.com/huggingface/trl.git
 
+
 # Step 6: Additional dependencies
 RUN uv pip install     gradio>=4.0.0     datasets>=2.18.0     anthropic>=0.39.0     huggingface_hub>=0.24.0     pyyaml     accelerate>=0.30.0     sentencepiece     protobuf
 
+
 # Set working directory
 WORKDIR /app
 
+
 # Copy application code
 COPY . /app/
 
+
 # Create directories for runs
 RUN mkdir -p /app/runs /app/data /app/outputs
 
+
 # Environment variables
 ENV PYTHONUNBUFFERED=1
 ENV HF_HUB_ENABLE_HF_TRANSFER=1
 ENV UNSLOTH_VLLM_STANDBY=1
 
+
 # Expose Gradio port
 EXPOSE 7860
 
+
 # Run app
 CMD ["python3", "app.py"]