|
--- |
|
license: mit |
|
tags: |
|
- pytorch |
|
- diffusers |
|
- text-to-image |
|
- diffusion-models-class |
|
--- |
|
|
|
#### Resource GitHub repo: (https://github.com/ShivamShrirao/diffusers/tree/main/examples/dreambooth) |
|
|
|
This model is a dreambooth diffusion model for text-to-image generation. |
|
|
|
## Usage |
|
|
|
```python |
|
from diffusers import StableDiffusionPipeline |
|
import torch |
|
|
|
pipe = StableDiffusionPipeline.from_pretrained('Gulsum06/diff-dreambooth-finetuning', torch_dtype=torch.float16) |
|
pipe = pipe.to("cuda") |
|
|
|
prompt = "photo of zwx dog in a bucket" |
|
image = pipe(prompt).images[0] |
|
|
|
image |
|
``` |
|
|