Likalto4 commited on
Commit
1b2e2aa
·
1 Parent(s): ebaefc7

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +22 -0
README.md ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - pytorch
5
+ - diffusers
6
+ - unconditional-image-generation
7
+ - diffusion-models-class
8
+ ---
9
+
10
+ This model is a diffusion model for unconditional image generation of mammograms of size 64x64 pixels.
11
+ The model was trained with 1000 images using the [DDPM](https://arxiv.org/abs/2006.11239) architecture.
12
+ The model was trained for 50 epochs with a batch size of 64, using around 11 GB of GPU memory.
13
+
14
+ ## Usage
15
+
16
+ ```python
17
+ from diffusers import DDPMPipeline
18
+
19
+ pipeline = DDPMPipeline.from_pretrained({hub_model_id})
20
+ image = pipeline().images[0]
21
+ image
22
+ ```