Update Dockerfile

Dockerfile

@@ -1,32 +1,13 @@
-# Use an official CUDA runtime with Ubuntu as a parent image
-FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
-
-# Install Python, pip, and git
-RUN apt-get update && apt-get install -y \
-    python3.11 \
-    python3-pip \
-    git \
-    && rm -rf /var/lib/apt/lists/*
-
-# Set the working directory in the container
-WORKDIR /usr/src/app
-
-# Clone the repository into the current directory
-RUN git clone https://github.com/theroyallab/tabbyAPI .
-
-# Install torch with CUDA support and exllamav2
-RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu121
-RUN pip install exllamav2
-
-# Install any other needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy the current directory contents into the container at /usr/src/app
-# Note: This might be redundant if you're cloning everything you need
-# COPY . /usr/src/app/
-
-# Make port 5000 available to the world outside this container
-EXPOSE 5000
-
-# Run main.py when the container launches
-CMD ["python3", "main.py"]
+FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
+RUN apt update && apt install git build-essential libopenblas-dev wget python3-pip -y
+RUN apt update && apt install -y wget
+ARG MODEL_PART_A
+ARG MODEL_PART_B
+ARG MODEL_NAME
+ARG ADDITIONAL
+RUN mkdir /opt/kobold
+RUN git clone https://github.com/0cc4m/KoboldAI -b latestgptq --recurse-submodules /opt/koboldcpp
+WORKDIR /opt/kobold
+RUN chmod +x install_requirements.sh
+RUN ./install_requirements.sh cuda
+CMD ./play.sh