remove the docker/ dir as this confuses everyone (including me)

docker/Dockerfile

@@ -1,83 +0,0 @@
-FROM nvidia/cuda:12.8.1-devel-ubuntu22.04
-
-LABEL authors="jaret"
-
-# Set noninteractive to avoid timezone prompts
-ENV DEBIAN_FRONTEND=noninteractive
-
-# ref https://en.wikipedia.org/wiki/CUDA
-ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0 10.0 12.0"
-
-# Install dependencies
-RUN apt-get update && apt-get install --no-install-recommends -y \
-    git \
-    curl \
-    build-essential \
-    cmake \
-    wget \
-    python3.10 \
-    python3-pip \
-    python3-dev \
-    python3-setuptools \
-    python3-wheel \
-    python3-venv \
-    ffmpeg \
-    tmux \
-    htop \
-    nvtop \
-    python3-opencv \
-    openssh-client \
-    openssh-server \
-    openssl \
-    rsync \
-    unzip \
-    && apt-get clean \
-    && rm -rf /var/lib/apt/lists/*
-
-# Install nodejs
-WORKDIR /tmp
-RUN curl -sL https://deb.nodesource.com/setup_23.x -o nodesource_setup.sh && \
-    bash nodesource_setup.sh && \
-    apt-get update && \
-    apt-get install -y nodejs && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
-
-WORKDIR /app
-
-# Set aliases for python and pip
-RUN ln -s /usr/bin/python3 /usr/bin/python
-
-# install pytorch before cache bust to avoid redownloading pytorch
-RUN pip install --pre --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
-
-# Fix cache busting by moving CACHEBUST to right before git clone
-ARG CACHEBUST=1234
-ARG GIT_COMMIT=main
-RUN echo "Cache bust: ${CACHEBUST}" && \
-    git clone https://github.com/ostris/ai-toolkit.git && \
-    cd ai-toolkit && \
-    git checkout ${GIT_COMMIT}
-
-WORKDIR /app/ai-toolkit
-
-# Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt && \
-    pip install --pre --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 --force && \
-    pip install setuptools==69.5.1 --no-cache-dir
-
-# Build UI
-WORKDIR /app/ai-toolkit/ui
-RUN npm install && \
-    npm run build && \
-    npm run update_db
-
-# Expose port (assuming the application runs on port 3000)
-EXPOSE 8675
-
-WORKDIR /
-
-COPY docker/start.sh /start.sh
-RUN chmod +x /start.sh
-
-CMD ["/start.sh"]

docker/start.sh

@@ -1,70 +0,0 @@
-#!/bin/bash
-set -e  # Exit the script if any statement returns a non-true return value
-
-# ref https://github.com/runpod/containers/blob/main/container-template/start.sh
-
-# ---------------------------------------------------------------------------- #
-#                          Function Definitions                                #
-# ---------------------------------------------------------------------------- #
-
-
-# Setup ssh
-setup_ssh() {
-    if [[ $PUBLIC_KEY ]]; then
-        echo "Setting up SSH..."
-        mkdir -p ~/.ssh
-        echo "$PUBLIC_KEY" >> ~/.ssh/authorized_keys
-        chmod 700 -R ~/.ssh
-
-         if [ ! -f /etc/ssh/ssh_host_rsa_key ]; then
-            ssh-keygen -t rsa -f /etc/ssh/ssh_host_rsa_key -q -N ''
-            echo "RSA key fingerprint:"
-            ssh-keygen -lf /etc/ssh/ssh_host_rsa_key.pub
-        fi
-
-        if [ ! -f /etc/ssh/ssh_host_dsa_key ]; then
-            ssh-keygen -t dsa -f /etc/ssh/ssh_host_dsa_key -q -N ''
-            echo "DSA key fingerprint:"
-            ssh-keygen -lf /etc/ssh/ssh_host_dsa_key.pub
-        fi
-
-        if [ ! -f /etc/ssh/ssh_host_ecdsa_key ]; then
-            ssh-keygen -t ecdsa -f /etc/ssh/ssh_host_ecdsa_key -q -N ''
-            echo "ECDSA key fingerprint:"
-            ssh-keygen -lf /etc/ssh/ssh_host_ecdsa_key.pub
-        fi
-
-        if [ ! -f /etc/ssh/ssh_host_ed25519_key ]; then
-            ssh-keygen -t ed25519 -f /etc/ssh/ssh_host_ed25519_key -q -N ''
-            echo "ED25519 key fingerprint:"
-            ssh-keygen -lf /etc/ssh/ssh_host_ed25519_key.pub
-        fi
-
-        service ssh start
-
-        echo "SSH host keys:"
-        for key in /etc/ssh/*.pub; do
-            echo "Key: $key"
-            ssh-keygen -lf $key
-        done
-    fi
-}
-
-# Export env vars
-export_env_vars() {
-    echo "Exporting environment variables..."
-    printenv | grep -E '^RUNPOD_|^PATH=|^_=' | awk -F = '{ print "export " $1 "=\"" $2 "\"" }' >> /etc/rp_environment
-    echo 'source /etc/rp_environment' >> ~/.bashrc
-}
-
-# ---------------------------------------------------------------------------- #
-#                               Main Program                                   #
-# ---------------------------------------------------------------------------- #
-
-
-echo "Pod Started"
-
-setup_ssh
-export_env_vars
-echo "Starting AI Toolkit UI..."
-cd /app/ai-toolkit/ui && npm run start