chinese-zodiac on stable diffuison by dreambooth

new concept

chinese-zodiac

inference

from torch import autocast
from diffusers import StableDiffusionPipeline
import torch
import diffusers
from PIL import Image


def image_grid(imgs, rows, cols):
    assert len(imgs) == rows*cols
    w, h = imgs[0].size
    grid = Image.new('RGB', size=(cols*w, rows*h))
    grid_w, grid_h = grid.size                    
    for i, img in enumerate(imgs):
        grid.paste(img, box=(i%cols*w, i//cols*h))
    return grid


pipe = StableDiffusionPipeline.from_pretrained("Dushwe/chinese-zodiac").to("cuda")
prompt = 'A mountain tip in the clouds,chinese-zodiac'
images = pipe(prompt, num_images_per_prompt=1, num_inference_steps=50, guidance_scale=7.5,torch_dtype=torch.cuda.HalfTensor).images
grid = image_grid(images, 1, 1)
grid

grid1

generate samples

crane under the tree,chinese-zodiac grid2 grid3

Chinese palace, 4k resolution, chinese-zodiac grid4

Moonlight,Mid-Autumn Festival,Moon,Silhouette,Lantern,White Rabbit,chinese-zodiac grid5

You run your new concept via diffusers Colab Notebook for Inference. Don't forget to use the concept prompts!