sweetfelinity
commited on
Commit
•
d9dce84
1
Parent(s):
c63cecd
Update README.md
Browse files
README.md
CHANGED
@@ -7,4 +7,48 @@ library_name: diffusers
|
|
7 |
tags:
|
8 |
- anime
|
9 |
- generative
|
10 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
tags:
|
8 |
- anime
|
9 |
- generative
|
10 |
+
---
|
11 |
+
|
12 |
+
# anime-faces-ddpm
|
13 |
+
A Denoising Diffusion Probabilistic Model (DDPM) trained to generate anime faces using [this example as a basis.](https://github.com/huggingface/diffusers/tree/main/examples/unconditional_image_generation)
|
14 |
+
|
15 |
+
![image/png](https://cdn-uploads.huggingface.co/production/uploads/65e6420da97628ed6ada1cb8/wM8CUMRQRoOV0FucyRpIi.png)
|
16 |
+
|
17 |
+
## Model Description
|
18 |
+
|
19 |
+
This diffusion model is trained with the [🤗 Diffusers](https://github.com/huggingface/diffusers) library
|
20 |
+
on the `huggan/anime-faces` dataset.
|
21 |
+
|
22 |
+
## How To Use
|
23 |
+
|
24 |
+
```python
|
25 |
+
from diffusers import DDPMPipeline
|
26 |
+
|
27 |
+
checkpoint = "sweetfelinity/anime-faces-ddpm"
|
28 |
+
pipeline = DDPMPipeline.from_pretrained(checkpoint)
|
29 |
+
pipeline = pipeline.to("cuda") # or "cpu"
|
30 |
+
|
31 |
+
for i in range(10):
|
32 |
+
image = pipeline().images[0]
|
33 |
+
image.save(str(i + 1) + ".png")
|
34 |
+
```
|
35 |
+
|
36 |
+
## Training Hyperparameters
|
37 |
+
|
38 |
+
The following hyperparameters were used during training:
|
39 |
+
- resolution=64
|
40 |
+
- train_batch_size=16
|
41 |
+
- num_epochs=30
|
42 |
+
- gradient_accumulation_steps=1
|
43 |
+
- learning_rate=1e-4
|
44 |
+
- lr_warmup_steps=500
|
45 |
+
- mixed_precision=fp16
|
46 |
+
- checkpointing_steps=2000
|
47 |
+
- save_images_epochs=4
|
48 |
+
- use_ema
|
49 |
+
- adam_weight_decay=1e-6
|
50 |
+
- lr_scheduler=linear
|
51 |
+
- eval_batch_size=32
|
52 |
+
|
53 |
+
## Training Results
|
54 |
+
See model folder /generated-images for 100 images created by the DDPM.
|