| language: | |
| - en | |
| tags: | |
| - stable-diffusion | |
| - pytorch | |
| - text-to-image | |
| - image-to-image | |
| - diffusion-models | |
| - computer-vision | |
| - generative-ai | |
| license: mit | |
| library_name: pytorch | |
| pipeline_tag: text-to-image | |
| base_model: stable-diffusion-v1-5 | |
| model-index: | |
| - name: pytorch-stable-diffusion | |
| results: | |
| - task: | |
| type: text-to-image | |
| name: Text-to-Image Generation | |
| dataset: | |
| type: custom | |
| name: Stable Diffusion v1.5 | |
| metrics: | |
| - type: inference_steps | |
| value: 50 | |
| - type: cfg_scale | |
| value: 8 | |
| - type: image_size | |
| value: 512x512 | |
| # pytorch-stable-diffusion | |
| PyTorch implementation of Stable Diffusion from scratch | |
| ## Download weights and tokenizer files: | |
| 1. Download `vocab.json` and `merges.txt` from https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/tree/main/tokenizer and save them in the `data` folder | |
| 2. Download `v1-5-pruned-emaonly.ckpt` from https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/tree/main and save it in the `data` folder | |
| ## Tested fine-tuned models: | |
| Just download the `ckpt` file from any fine-tuned SD (up to v1.5). | |
| 1. InkPunk Diffusion: https://huggingface.co/Envvi/Inkpunk-Diffusion/tree/main | |
| 2. Illustration Diffusion (Hollie Mengert): https://huggingface.co/ogkalu/Illustration-Diffusion/tree/main | |
| ## Features: | |
| - Text-to-image generation | |
| - Image-to-image generation | |
| - Support for fine-tuned models | |
| - PyTorch implementation from scratch |