File size: 1,178 Bytes
5f49b15 |
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 54 55 56 57 58 59 60 61 62 63 |
---
license: mit
tags:
- diffusion
- anime
- image-generation
- pytorch
---
# Anime Face Diffusion Model
A diffusion model trained to generate 64x64 anime-style faces.
## Model Details
- **Model type**: Denoising Diffusion Probabilistic Model (DDPM)
- **Training data**: Anime faces dataset
- **Image size**: 64x64 RGB
- **Timesteps**: 1000
- **Architecture**: U-Net with sinusoidal time embeddings
## Usage
```python
import torch
from diffusion_model import SimpleUnet, Diffuser # Your model code
# Load the model
checkpoint = torch.load("model.pt")
model = SimpleUnet(in_channels=3)
model.load_state_dict(checkpoint['ema_model_state_dict']) # Use EMA weights
model.eval()
# Generate samples
# (Add your sampling code here)
```
## Training Details
- Trained for X epochs
- Batch size: 64
- Learning rate: 1e-4
- EMA decay: 0.9999
- Noise schedule: Cosine
## Samples
(Add generated sample images here)
## Citation
If you use this model, please cite:
```
@misc{your-anime-diffusion,
author = {Your Name},
title = {Anime Face Diffusion Model},
year = {2024},
publisher = {HuggingFace},
url = {https://huggingface.co/your-username/anime-diffusion}
}
```
|