Upload Dockerfile with huggingface_hub

Dockerfile

@@ -5,18 +5,22 @@ WORKDIR /app
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
-    && rm -rf /var/lib/apt/lists/*
-
-# Copy requirements
-COPY requirements_v5.txt requirements.txt
-
-# Install Python dependencies
-# Note: Some packages may fail - that's OK, script has fallbacks
-RUN pip install --no-cache-dir torch transformers accelerate trl datasets huggingface_hub peft bitsandbytes optimum || \
-    pip install --no-cache-dir torch transformers accelerate trl datasets huggingface_hub peft optimum
-
-# Try to install IPEX (may fail on non-Intel)
-RUN pip install --no-cache-dir intel_extension_for_pytorch || echo "IPEX not installed (non-Intel CPU)"
+    && rm -rf /var/lib/aptLists/*
+
+# Install core Python packages (required)
+RUN pip install --no-cache-dir \
+    torch \
+    transformers \
+    accelerate \
+    trl \
+    datasets \
+    huggingface_hub \
+    peft
+
+# Install optional packages (may fail, that's OK)
+RUN pip install --no-cache-dir optimum || echo "optimum not installed"
+RUN pip install --no-cache-dir bitsandbytes || echo "bitsandbytes not installed"
+RUN pip install --no-cache-dir intel_extension_for_pytorch || echo "IPEX not installed"
 
 # Copy training script
 COPY train_arithmetic_v5_ultimate.py .