ntkuhn commited on
Commit
5f49b15
·
verified ·
1 Parent(s): 968a510

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +62 -0
  2. model.pt +3 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - diffusion
5
+ - anime
6
+ - image-generation
7
+ - pytorch
8
+ ---
9
+
10
+ # Anime Face Diffusion Model
11
+
12
+ A diffusion model trained to generate 64x64 anime-style faces.
13
+
14
+ ## Model Details
15
+
16
+ - **Model type**: Denoising Diffusion Probabilistic Model (DDPM)
17
+ - **Training data**: Anime faces dataset
18
+ - **Image size**: 64x64 RGB
19
+ - **Timesteps**: 1000
20
+ - **Architecture**: U-Net with sinusoidal time embeddings
21
+
22
+ ## Usage
23
+
24
+ ```python
25
+ import torch
26
+ from diffusion_model import SimpleUnet, Diffuser # Your model code
27
+
28
+ # Load the model
29
+ checkpoint = torch.load("model.pt")
30
+ model = SimpleUnet(in_channels=3)
31
+ model.load_state_dict(checkpoint['ema_model_state_dict']) # Use EMA weights
32
+ model.eval()
33
+
34
+ # Generate samples
35
+ # (Add your sampling code here)
36
+ ```
37
+
38
+ ## Training Details
39
+
40
+ - Trained for X epochs
41
+ - Batch size: 64
42
+ - Learning rate: 1e-4
43
+ - EMA decay: 0.9999
44
+ - Noise schedule: Cosine
45
+
46
+ ## Samples
47
+
48
+ (Add generated sample images here)
49
+
50
+ ## Citation
51
+
52
+ If you use this model, please cite:
53
+
54
+ ```
55
+ @misc{your-anime-diffusion,
56
+ author = {Your Name},
57
+ title = {Anime Face Diffusion Model},
58
+ year = {2024},
59
+ publisher = {HuggingFace},
60
+ url = {https://huggingface.co/your-username/anime-diffusion}
61
+ }
62
+ ```
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:828faa2a6723ea2659ee98ccafaaa121162379383a782ed3d056726999ea6dc7
3
+ size 724380479