Description

Stable Diffusion inference using HuggingFace diffusers library with an interactive run on VESSL.

YAML

name: stable-diffusion
description: "Stable Diffusion inference using HuggingFace diffusers library with an interactive run on VESSL."
resources:
  cluster: aws-apne2
  preset: v1.v100-1.mem-52
image: quay.io/vessl-ai/ngc-pytorch-kernel:22.12-py3-202301160809
import:
  /root/examples: git://github.com/vessl-ai/examples/
run:
  - workdir: /root/examples/stable-diffusion
    command: |
      bash ./setup.sh && streamlit run main.py
interactive:
  max_runtime: 24h
  jupyter:
    idle_timeout: 120m
ports:
  - name: streamlit
    type: http
    port: 8501

Demo