# Stable Diffusion in pytorch

A single file of Stable Diffusion. It is simple, easy reader.I hope you enjoyed. I hope you can discovery light!!!

The weights were ported from the original implementation.


## Usage

### Download weights .pt file and clone project

#### weights file

1. sd-v1-4.ckpt(4GB) https://drive.google.com/file/d/13XKPH-RdQ-vCvaJJgVR7W6q9R5XbaTLM/view?usp=share_link
2. v1-5-pruned.ckpt(4GB, not include ema weights) https://drive.google.com/file/d/1IwBQ0DWfSNA50ymBvY0eby7v9RSIdSWu/view?usp=share_link
3. mdjrny-v4.ckpt(4GB, some weights cast float16 to float32) https://drive.google.com/file/d/1-Z5bE9GBpuupuyhoXWFZiEtldBzVJ61X/view?usp=share_link
4. waifu-diffusion-v1-4 weight
5. animev3.pt
6. Anything-V3.0.pt
7. 4,5,6 and other down address is https://huggingface.co/xfh/min-stable-diffusion-pt/tree/main

#### clone project

```bash
git clone https://github.com/scale100xu/min-stable-diffusion.git
```

#### Using pip install

Install dependencies using the `requirements.txt` file:

```bash
pip install -r requirements.txt
```

### help

```bash
python stable_diffusion.py --help
```

```
usage: stable_diffusion.py [-h] [--steps STEPS] [--phrase PHRASE] [--out OUT] [--scale SCALE] [--model_file MODEL_FILE] [--img_width IMG_WIDTH] [--img_height IMG_HEIGHT] [--seed SEED]
                           [--device_type DEVICE_TYPE]

Run Stable Diffusion

options:
  -h, --help            show this help message and exit
  --steps STEPS         Number of steps in diffusion (default: 25)
  --phrase PHRASE       Phrase to render (default: anthropomorphic cat portrait art )
  --out OUT             Output filename (default: /tmp/rendered.png)
  --scale SCALE         unconditional guidance scale (default: 7.5)
  --model_file MODEL_FILE
                        model weight file (default: /tmp/stable_diffusion_v1_4.pt)
  --img_width IMG_WIDTH
                        output image width (default: 512)
  --img_height IMG_HEIGHT
                        output image height (default: 512)
  --seed SEED           random seed (default: 443)
  --device_type DEVICE_TYPE
                        random seed (default: cpu)

```
### Using `stable_diffusion.py` from the git repo

Assuming you have installed the required packages, 
you can generate images from a text prompt using:

```bash
python stable_diffusion.py --model_file="/tmp/stable_diffusion_v1_4.pt" --phrase="An astronaut riding a horse" --device_type="cuda"
```

The generated image will be named `/tmp/render.png` on the root of the repo.
If you want to use a different name, use the `--out` flag.

```bash
python stable_diffusion.py --model_file="/tmp/stable_diffusion_v1_4.pt"  --phrase="An astronaut riding a horse" --out="/tmp/image.png" --device_type="cuda"
```

## Example outputs 

The following outputs have been generated using this implementation:

1) anthropomorphic cat portrait art

![a](https://huggingface.co/spaces/xfh/min-stable-diffusion-web/resolve/main/rendered.png)

2) anthropomorphic cat portrait art(mdjrny-v4.pt)

![a](https://huggingface.co/spaces/xfh/min-stable-diffusion-web/resolve/main/rendered2.png)

3) Kung Fu Panda(weight: wd-1-3-penultimate-ucg-cont.pt, steps:50)

![a](https://huggingface.co/spaces/xfh/min-stable-diffusion-web/resolve/main/rendered3.png)
![a](https://huggingface.co/spaces/xfh/min-stable-diffusion-web/resolve/main/rendered4.png)



## References

1) https://github.com/CompVis/stable-diffusion
2) https://github.com/geohot/tinygrad/blob/master/examples/stable_diffusion.py