mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-18 07:41:03 +00:00
Image Gen Dockerfile and Github workflow
This commit is contained in:
@@ -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
|
||||
}
|
||||
Reference in New Issue
Block a user