Guohua-Diffusion / README.md
wangyulong's picture
Update README.md
d870a95
|
raw
history blame
1.11 kB
metadata
license: creativeml-openrail-m
tags:
  - stable-diffusion
  - text-to-image

Guohua Diffusion

This is the fine-tuned Stable Diffusion model trained on traditional Chinese paintings.

Use guohua style in your prompts for the effect.

Sample Image

example1 example2

How to use

WebUI

Download the guohua.ckpt in model files.

Diffusers

This model can be used just like any other Stable Diffusion model. For more information, please have a look at the Stable Diffusion.

#!pip install diffusers transformers scipy torch
from diffusers import StableDiffusionPipeline
import torch

model_id = "Langboat/Guohua-Diffusion"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")

prompt = "The Godfather poster in guohua style"
image = pipe(prompt).images[0]

image.save("./the_god_father.png")