johnowhitaker commited on
Commit
b9d1951
1 Parent(s): 40d18c2

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +24 -0
README.md ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ This model is a diffusion model initialized from https://huggingface.co/google/ddpm-bedroom-256 and trained for 5000 steps on https://huggingface.co/datasets/huggan/wikiart.
13
+ Script: https://github.com/huggingface/diffusion-models-class/blob/main/unit2/finetune_model.py
14
+ Training Logs (with example images): https://wandb.ai/johnowhitaker/dm_finetune/runs/2upaa341
15
+
16
+ ## Usage
17
+
18
+ ```python
19
+ from diffusers import DDPMPipeline
20
+
21
+ pipeline = DDPMPipeline.from_pretrained('johnowhitaker/sd-class-wikiart-from-bedrooms')
22
+ image = pipeline().images[0]
23
+ image
24
+ ```