File size: 1,534 Bytes
f5b74e6 f2e3711 045bf19 |
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 |
---
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 |