Upload Dockerfile with huggingface_hub

Dockerfile

@@ -20,30 +20,29 @@ ENV PATH="/home/user/.local/bin:${PATH}"
 WORKDIR /home/user/app
 
 # ============================================================
-# FIXED Installation Order - Critical for unsloth compatibility
+# CRITICAL: Version pinning for unsloth compatibility
 # ============================================================
 
 # a. Upgrade pip first
-RUN python3 -m pip install --upgrade pip
+RUN python3 -m pip install --upgrade pip setuptools wheel
 
-# b. Install torch with CUDA 12.1 (MUST come first)
-RUN python3 -m pip install torch==2.3.1+cu121 torchvision==0.18.1+cu121 torchaudio==2.3.1+cu121 --index-url https://download.pytorch.org/whl/cu121
+# b. Install torch with CUDA 12.1 (MUST be 2.3.1 - 2.4.0 breaks unsloth_zoo)
+RUN python3 -m pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu121
 
-# c. Install transformers (MUST be 4.44.x - 4.46.x for unsloth_zoo)
+# c. Install transformers (MUST be 4.44.2 - 4.46+ breaks Unpack import)
 RUN python3 -m pip install transformers==4.44.2
 
-# d. Install unsloth (core only, not the git version)
+# d. Install xformers (compatible version)
+RUN python3 -m pip install xformers==0.0.27
+
+# e. Install unsloth AFTER torch and transformers
 RUN python3 -m pip install unsloth
 
-# e. Install unsloth_zoo (required dependency)
+# f. Install unsloth_zoo (MUST be after unsloth)
 RUN python3 -m pip install unsloth_zoo
 
-# f. Install other required packages
-RUN python3 -m pip install xformers==0.0.27 peft accelerate bitsandbytes datasets huggingface_hub wandb trl
-
-# g. Install any remaining requirements
-COPY --chown=user:user requirements.txt .
-RUN python3 -m pip install -r requirements.txt || true
+# g. Install other required packages
+RUN python3 -m pip install peft accelerate bitsandbytes datasets huggingface_hub wandb trl openai pydantic pyyaml fastapi uvicorn
 
 # Copy application code
 COPY --chown=user:user train.py .