README.md

metadata

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
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