---
license: creativeml-openrail-m
library_name: diffusers
language:
- en
tags:
- 'dreambooth '
- 'diffusion '
- transformers
pipeline_tag: text-to-image
---

Dreambooth style: Avatar
Dreambooth finetuning of Stable Diffusion (v1.5.1) on Avatar art style.

About
This text-to-image stable diffusion model was trained with dreambooth.
Put in a text prompt and generate your own Avatar style image!

![pk1.jpg](https://staticassetbucket.s3.us-west-1.amazonaws.com/avatar_grid.png)
(Image taken from Lambdalabs repo)

```
from diffusers import DiffusionPipeline, UniPCMultistepScheduler
import torch
from torch import autocast
pipeline = DiffusionPipeline.from_pretrained(
    "Andyrasika/avatar_diffusion",
    custom_pipeline="lpw_stable_diffusion",

    torch_dtype=torch.float16
)
pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)
pipeline.to("cuda")

pipeline.enable_vae_tiling()
pipeline.enable_xformers_memory_efficient_attention()

prompt = "Yoda, avatarart style"
scale = 7.5
n_samples = 4

with autocast("cuda"):
  images = pipeline(n_samples*[prompt], guidance_scale=scale).images

for idx, im in enumerate(images):
  im.save(f"{idx:06}.png")
```