# Dockerfile Public A10G

# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.2.0/ubuntu2204/base/Dockerfile
FROM nvidia/cuda:12.2.0-base-ubuntu22.04
ENV DEBIAN_FRONTEND noninteractive

RUN apt-get update -y \
	apt-get install -y aria2 libgl1 libglib2.0-0 wget git git-lfs python3-pip python-is-python3 \
	pip install -q torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118 torchtext==0.15.2 torchdata==0.6.1 --extra-index-url https://download.pytorch.org/whl/cu118 \
	pip install xformers==0.0.20 triton==2.0.0 \
	adduser --disabled-password --gecos '' user \
	mkdir /content \
	chown -R user:user /content \
 	git clone -b v2.6 https://github.com/camenduru/stable-diffusion-webui /content/stable-diffusion-webui \
	sed -i -e 's/    start()/    #start()/g' /content/stable-diffusion-webui/launch.py \
	python /content/stable-diffusion-webui/launch.py --skip-torch-cuda-test \
	git -C /content/stable-diffusion-webui reset --hard \
	wget https://github.com/camenduru/webui-docker/raw/main/header_patch.py -O /content/header_patch.py \
	sed -i -e '/demo:/r /content/header_patch.py' /content/stable-diffusion-webui/modules/ui.py \
	aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/juggernaut-xl/resolve/main/juggernautXL_version2.safetensors -d /content/stable-diffusion-webui/models/Stable-diffusion -o juggernautXL_version2.safetensors

WORKDIR /content
USER user
CMD python /content/stable-diffusion-webui/launch.py --xformers --listen