Update Dockerfile

Dockerfile

@@ -1,31 +1,29 @@
-# Use an appropriate base image with Python and CUDA support if needed
-FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 
+# Use a base image with Python
+FROM python:3.10-slim-buster 
 
-# Set working directory
-WORKDIR /app
+# Set working directory
+WORKDIR /app
 
-# Copy requirements file
+# Copy requirements file
+COPY requirements.txt .
+COPY extensions/openai/requirements.txt .
 
-COPY requirements.txt .
-COPY extensions/openai/requirements.txt .
+# Install dependencies
+RUN pip install --upgrade pip \
+    && pip install -r extensions/openai/requirements.txt \
+    && pip install -r requirements.txt
 
-# Install dependencies
-RUN pip install --upgrade pip \
-    && pip install -r extensions/openai/requirements.txt \
-    && pip install -r requirements.txt
+# Clone the repository
+RUN git clone https://github.com/oobabooga/text-generation-webui
 
-# Clone the repository
+# Set working directory to the cloned repository
+WORKDIR /app/text-generation-webui
 
-RUN git clone https://github.com/oobabooga/text-generation-webui
+# Download the model 
+RUN python download-model.py https://huggingface.co/DanielAWrightGabrielAI/mpt-7b-storywriter-4bit-128g-65kTokens-CPU
 
-# Set working directory to the cloned repository
-WORKDIR /app/text-generation-webui
+# Expose the port the web UI will run on
+EXPOSE 7860
 
-# Download the model 
-RUN python download-model.py https://huggingface.co/DanielAWrightGabrielAI/mpt-7b-storywriter-4bit-128g-65kTokens-CPU
-
-# Expose the port the web UI will run on
-EXPOSE 7860
-
-# Define the command to run the web UI 
-CMD ["python", "server.py", "--share", "--model", "/app/text-generation-webui/models/DanielAWrightGabrielAI_mpt-7b-storywriter-4bit-128g-65kTokens-CPU", "--tensorcores", "--trust-remote-code", "--api", "--public-api"
+# Define the command to run the web UI (remove --tensorcores flag)
+CMD ["python", "server.py", "--share", "--model", "/app/text-generation-webui/models/DanielAWrightGabrielAI_mpt-7b-storywriter-4bit-128g-65kTokens-CPU", "--trust-remote-code", "--api", "--public-api"