Update Dockerfile

Dockerfile

@@ -33,8 +33,15 @@ ARG PYTHON_VERSION=3.9.17
 RUN pyenv install $PYTHON_VERSION && \
     pyenv global $PYTHON_VERSION && \
     pyenv rehash && \
-    pip install --no-cache-dir --upgrade pip setuptools wheel && \
-    pip install --no-cache-dir \
+    #pip install --no-cache-dir --upgrade pip setuptools wheel && \
+   # pip install --no-cache-dir \
+    # ... (giữ nguyên phần trên của Dockerfile)
+
+# Cài đặt requirements với nguồn PyTorch bổ sung
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
+    pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu118 -r /code/requirements.txt
+
+# ... (giữ nguyên phần dưới)
     datasets \
     huggingface-hub "protobuf<4" "click<8.1"
 	#pip install huggingface_hub["cli"] && \