enzokro commited on
Commit
75174ee
1 Parent(s): 5b5ddbe

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +26 -0
README.md ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - pytorch
5
+ - diffusers
6
+ - unconditional-image-generation
7
+ - diffusion-models-class
8
+ ---
9
+
10
+ # Model Card for Unit 1 of the [Diffusion Models Class 🧨](https://github.com/huggingface/diffusion-models-class)
11
+
12
+ This model is a diffusion model for unconditional image generation of cute 🦋.
13
+
14
+ The model was trained with Adam parameters from fast.ai.
15
+ Batch size was also doubled to 64.
16
+ Learning rate happens over 160 steps, aka 20% of training.
17
+
18
+ ## Usage
19
+
20
+ ```python
21
+ from diffusers import DDPMPipeline
22
+
23
+ pipeline = DDPMPipeline.from_pretrained('enzokro/sd-class-butterflies-64')
24
+ image = pipeline().images[0]
25
+ image
26
+ ```