File size: 1,188 Bytes
dddb394 bf4e115 495b168 628853a 495b168 ea17913 495b168 bf4e115 429d78c bf4e115 495b168 ea17913 495b168 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
---
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!
(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")
``` |