patrickvonplaten commited on
Commit
0bd77de
1 Parent(s): 1e80c0c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -1,9 +1,14 @@
 
 
 
 
 
1
  # Dummy diffusion model following architecture of https://github.com/lucidrains/denoising-diffusion-pytorch
2
 
3
  Run the model as follows:
4
 
5
  ```python
6
- from diffusers import UNetModel
7
  import torch
8
 
9
  model = UNetModel.from_pretrained("fusing/ddpm_dummy")
1
+ ---
2
+ tags:
3
+ - hf_diffuse
4
+ ---
5
+
6
  # Dummy diffusion model following architecture of https://github.com/lucidrains/denoising-diffusion-pytorch
7
 
8
  Run the model as follows:
9
 
10
  ```python
11
+ from diffusers import UNetModel, GaussianDiffusion
12
  import torch
13
 
14
  model = UNetModel.from_pretrained("fusing/ddpm_dummy")