fix: updated torch base

Dockerfile

@@ -1,25 +1,10 @@
 # Base image
-FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04
-
-ENV DEBIAN_FRONTEND=noninteractive
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV PYTHONUNBUFFERED=1
-
-# Use bash shell with pipefail option
-SHELL ["/bin/bash", "-o", "pipefail", "-c"]
-
-# Set the working directory
-WORKDIR /
-
-# Update and upgrade the system packages (Worker Template)
-COPY builder/setup.sh /setup.sh
-RUN /bin/bash /setup.sh && \
-    rm /setup.sh
+FROM runpod/base:0.4.2-cuda11.8.0
 
 # Install Python dependencies (Worker Template)
 COPY builder/requirements.txt /requirements.txt
-RUN python3 -m pip install --upgrade pip && \
-    python3 -m pip install --upgrade -r /requirements.txt --no-cache-dir && \
+RUN python3.11 -m pip install --upgrade pip && \
+    python3.11 -m pip install --upgrade -r /requirements.txt --no-cache-dir && \
     rm /requirements.txt
 
 # Cache Models
@@ -30,4 +15,4 @@ RUN python3 /cache_models.py && \
 # Add src files (Worker Template)
 ADD src .
 
-CMD python3 -u /rp_handler.py
+CMD python3.11 -u /rp_handler.py

builder/requirements.txt

@@ -1,7 +1,7 @@
 # Required Python packages get listed here, one per line.
 # Reccomended to lock the version number to avoid unexpected changes.
 
-runpod==1.3.0
+runpod==1.4.0
 
 diffusers==0.21.4
 transformers==4.34.0
@@ -13,4 +13,4 @@ PyWavelets>=1.1.1
 opencv-python>=4.8.1.78
 invisible-watermark==0.2.0
 xformers==0.0.22
-torch==2.0.1+cu118
+torch==2.1.1+cu118