RepoReplicateReplicatepublished Sep 22, 2023seen 5d

replicate/dreambooth-batch

Python

Open original ↗

Captured source

source ↗
published Sep 22, 2023seen 5dcaptured 8hhttp 200method plain

replicate/dreambooth-batch

Description: batch inference (image generation) from dreambooth trainings

Language: Python

License: Apache-2.0

Stars: 0

Forks: 1

Open issues: 0

Created: 2023-09-22T20:37:38Z

Pushed: 2023-09-23T00:19:18Z

Default branch: main

Fork: no

Archived: no

README:

Stable Diffusion Dreambooth model template

This is the template used by replicate.com dreambooth api to build custom dreambooth models.

This template is based on cog-stable-diffusion which uses diffusers.

Usage

This template is primarily intended for use by Replicate's DreamBooth API, which is what you probably want to use to train and publish your own model.

If you really want to use this template locally, you can do so by following these steps:

1. Generate some weights, put them in weights/ (use our trainer or your own) 2. Download NSFW safety_checker weights using script/download-weights 3. Install cog & docker 4. Build cog build 5. Predict cog predict -i prompt="photo of zzz" -i seed=42 ...