Image Gen Dockerfile and Github workflow

This commit is contained in:
ArEnSc
2024-02-02 00:53:20 -05:00
parent bcb9797246
commit 335d906547
3 changed files with 124 additions and 0 deletions
@@ -0,0 +1,39 @@
# Use NVIDIA CUDA with Ubuntu as the base image
FROM nvidia/cuda:12.2.0-devel-ubuntu20.04
# Set environment variables
ENV MY_ENV_VARIABLE=value
ENV DEBIAN_FRONTEND=noninteractive
RUN mkdir /python_install
RUN mkdir /model_code
WORKDIR /python_install
# Install necessary packages
RUN apt-get update && apt-get install -y \
curl \
software-properties-common \
git \
ffmpeg \
espeak-ng \
sudo
# rsync to copy between pods with rich logs
RUN apt-get install -y rsync
RUN add-apt-repository ppa:deadsnakes/ppa
RUN apt-get update
RUN apt-get install -y python3.10 python3.10-venv python3.10-dev python3.10-full python3.10-distutils
COPY requirements.txt .
# Install Python dependencies
RUN python3.10 -m venv python \
&& python/bin/pip install --upgrade pip wheel setuptools \
&& . python/bin/activate \
&& pip install -r requirements.txt
WORKDIR /model_code
COPY . .
@@ -0,0 +1,19 @@
{
"uuid_idempotency_token": "11",
"type_of_inference": "inference",
"maybe_sd_model_token": "weight_dmmthavhawqc2hj7yqyemcbf8",
"maybe_lora_model_token": "weight_t7gz78fjg27m0wtw6r33gafxs",
"maybe_prompt": "raiden mei a very well drawn, anime girl, with pink and purple hair sitting down on a chair relaxed, highest quality, semi nude, masterpiece, painted",
"maybe_a_prompt": "a anime girl, with pink and purple hair sitting down on a chair relaxed.",
"maybe_n_prompt": "nsfw, black and white, low quality, pixelated",
"maybe_seed": -1,
"maybe_width": 1024,
"maybe_height": 1024,
"maybe_sampler": "DPM++ 2M SDE Heun",
"maybe_upload_path": "https://drive.google.com/file/d/1WRgR2pn0Ky8ls5_9Zq6tQlHTBvyWeach/view?usp=sharing",
"maybe_lora_upload_path": "https://drive.google.com/file/d/1WRgR2pn0Ky8ls5_9Zq6tQlHTBvyWeach/view?usp=sharing",
"maybe_cfg_scale": 7,
"maybe_number_of_samples": 64,
"maybe_batch_count": 4
}