runpod / serverless-ckpt-template

🖼️ | Quickly deploy a custom RunPod Endpoint API using your own model ckpt.
MIT License
21 stars 3 forks source link

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:

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](https://www.runpod.io/console/serverless/user/templates) on RunPod.

![template setup](./docs/images/template-setup.png)

![API Setup](./docs/images/api-setup.png)

![Endpoint](./docs/images/endpoint.png)

## Usage

To use your endpoint see our [Endpoint Docs](https://docs.runpod.io/ai-endpoints/runpod-apis)