update dockerfile

Dockerfile

@@ -41,7 +41,9 @@ RUN pyenv install ${PYTHON_VERSION} && \
     pyenv rehash && \
     pip install --no-cache-dir -U pip setuptools wheel
 
-RUN pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1
+# RUN pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1
+RUN pip install --no-cache-dir -U torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117
+RUN pip install --no-cache-dir -U xformers==0.0.17
 COPY --chown=1000 requirements.txt /tmp/requirements.txt
 RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
 RUN pip install -e Make-A-Protagonist/experts/GroundedSAM/segment_anything

requirements.txt

@@ -12,8 +12,8 @@ omegaconf==2.3.0
 Pillow==9.4.0
 python-slugify==7.0.0
 tensorboard==2.11.2
-torch==1.13.1
-torchvision==0.14.1
+# torch==1.13.1
+# torchvision==0.14.1
 transformers==4.27.4
 triton==2.0.0.post1
-xformers==0.0.17
+# xformers==0.0.17