sweetfelinity commited on
Commit
d9dce84
1 Parent(s): c63cecd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +45 -1
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.