Hugging Face's logo

wx44wx
/
three-kingdoms-stable-diffusion

Text-to-Image
English
stable-diffusion
stable-diffusion-diffusers
Model card Files Community
three-kingdoms-stable-diffusion
File size: 2,165 Bytes 
09b2f5a
 
590eb27
 
 
 
 
 
 
 
09b2f5a
590eb27
 
 
 
 
be8954f
590eb27
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c844e7c
 
 
 
 
be8954f
c844e7c
590eb27
 
 
 
 
 
 
 
 
 
 
 
 
 
35c4fbb
 
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
49
50
51
52
53
54
55
56
57
 
---
license: mit
language: 
  - en
tags:
- stable-diffusion
- stable-diffusion-diffusers
- text-to-image
datasets:
- wx44wx/three-kingdoms-blip-captions
---

__Stable Diffusion fine tuned on [Romance of the Three Kingdoms XI: Officer Portraits](https://kongming.net/11/portraits/).__

Put in a text prompt and generate your own Officier in Three Kingdoms.

trained using this [script](https://github.com/WangXin93/three-kingdoms-stable-diffusion) with this [dataset](https://huggingface.co/datasets/wx44wx/three-kingdoms-blip-captions).

> a man in armor
![image.png](https://github.com/WangXin93/three-kingdoms-stable-diffusion/raw/main/assets/a-man-in-armor.png)

> a women in red dress
![image.png](https://github.com/WangXin93/three-kingdoms-stable-diffusion/raw/main/assets/a-women-in-red-dress.png)

> a women in armor
 ![image.png](https://github.com/WangXin93/three-kingdoms-stable-diffusion/raw/main/assets/a-women-in-armor.png)

## Usage

```bash
!pip install diffusers==0.19.3
!pip install transformers scipy ftfy
```

```python
 python scripts/txt2img.py \
    --prompt 'a man in armor' \
    --outdir 'outputs/generated_three_kingdoms_officers' \
    --H 512 --W 512 \
    --n_samples 4 \
    --config 'configs/stable-diffusion/three-kingdoms.yaml' \
    --ckpt ema-only-epoch=000144.ckpt
```

## Model description

Trained on [BLIP captioned Three Kingdoms Officers images](https://huggingface.co/datasets/wx44wx/three-kingdoms-blip-captions) using 1xA6000 GPUs for around 16,000 steps.

## Links

- [Lambda Diffusers](https://github.com/LambdaLabsML/lambda-diffusers)
- [Captioned Three Kingdoms dataset](https://huggingface.co/datasets/wx44wx/three-kingdoms-blip-captions)
- [Model weights in Diffusers format](https://huggingface.co/wx44wx/sd-three-kingdoms-diffusers)
- [Original model weights](https://huggingface.co/wx44wx/three-kingdoms-stable-diffusion)
- [Training code](https://github.com/justinpinkney/stable-diffusion)

Trained by [Xin Wang](wangxin93.github.io). Thanks [kongming.net](kongming.net) for their archived images and [justinpinkney](https://github.com/LambdaLabsML/examples/tree/main/stable-diffusion-finetuning) for the code.