Skier8402 commited on
Commit
8858ec5
1 Parent(s): 0cba846

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
+ # Example Fine-Tuned Model for Unit 2 of the [Diffusion Models Class 🧨](https://github.com/huggingface/diffusion-models-class)
11
+
12
+ Fine-tuning the smithosonian butterflies dataset with celebhq dataset. --image_size 128 --batch_size 8 --num_epochs 16 --grad_accumulation_steps 2
13
+
14
+ ## Usage
15
+
16
+ ```python
17
+ from diffusers import DDPMPipeline
18
+
19
+ pipeline = DDPMPipeline.from_pretrained('Skier8402/ddpm-celebahq-finetuned-butterflies-16epochs')
20
+ image = pipeline().images[0]
21
+ image
22
+ ```