Update Dockerfile

Dockerfile

@@ -10,13 +10,17 @@ WORKDIR /workspace
 
 # Upgrade pip and install CPU-only PyTorch, Transformers, Accelerate, Unsloth, etc.
 # Use the official PyTorch CPU wheel index for performance on CPU:contentReference[oaicite:5]{index=5}.
-RUN pip3 install --upgrade pip \
- && pip3 install torch --index-url https://download.pytorch.org/whl/cpu \
- && pip3 install transformers accelerate unsloth
+
+#RUN pip3 install --upgrade pip \
+# && pip3 install torch --index-url https://download.pytorch.org/whl/cpu \
+# && pip3 install transformers accelerate unsloth
 
 # (Optional) Install unsloth_zoo or other utilities if needed:
 # RUN pip3 install unsloth-zoo
 
+RUN pip3 install --upgrade pip \
+ && pip3 install transformers accelerate unsloth
+
 # Copy an example inference script into the container
 # (This script should load the model and do a sample generation.)
 COPY inference.py /workspace/inference.py