NanoDiffusion-46M / README.md
FlameF0X's picture
Update README.md
cafa332 verified
|
Raw
History Blame Contribute Delete
776 Bytes
---
tags:
- text-to-image
- diffusion
- latent-diffusion
- pytorch
license: apache-2.0
---
# NanoDiffusion-46M
A ~46M parameter UNet trained from scratch for text-to-image generation
in latent space (SD VAE + CLIP text encoder, both frozen).
**Dataset:** [BitTranslate/Bittensor_subnet_19_06_04_24](https://huggingface.co/datasets/BitTranslate/Bittensor_subnet_19_06_04_24)
**Trained steps:** 500
**Image size:** 256px → 32×32 latents
## Architecture
- **UNet base channels:** 128
- **Channel multipliers:** (1, 2, 3, 4)
- **Res blocks per level:** 2
- **Cross-attention (text conditioning):** latent res (8,)
- **VAE (frozen):** `runwayml/stable-diffusion-v1-5` vae subfolder
- **Text encoder (frozen):** CLIP ViT-L/14 from `runwayml/stable-diffusion-v1-5`