Likalto4 commited on
Commit
8e3dfe0
·
1 Parent(s): dfd3472

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 is a latent diffusion model for generation of mammograms with size 64x64 in the latent space.
11
+ It was trained with 4K images in the latent space (10% of the original dataset) for 50 epochs.
12
+ Images in the latent space have size (4,64,64) and can generate images of 512x512 when decoded using the VAE.
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
+ ```