Skip to content

runpod/serverless-ckpt-template

Repository files navigation

Serverless | Model Checkpoint Template

This repository provides an end-to-end template for deploying your own Stable Diffusion Model to RunPod Serverless. The setup scripts will help to download the model and set up the Dockerfile.

Setup

You will need the following:

  • Image repository (e.g., Docker Hub)
  • RunPod account
  • Selected model from HuggingFace
  • S3 bucket (optional)
git clone https://github.com/runpod/serverless-ckpt-template.git
cd serverless-ckpt-template

docker build --build-arg MODEL_URL={huggingface.co/model/id} -t repo/image_name:tag .
docker push repo/image_name:tag

Once you have pushed your Docker image, navigate to the Serverless Templates on RunPod.

template setup

API Setup

Endpoint

Usage

To use your endpoint see our Endpoint Docs

About

🖼️ | Quickly deploy a custom RunPod Endpoint API using your own model ckpt.

Resources

License

Stars

Watchers

Forks