lysandre HF staff commited on
Commit
65758ed
1 Parent(s): 42719dd

Add diffusion model

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. config.json +1 -0
  3. diffusion_model.pt +3 -0
.gitattributes CHANGED
@@ -25,3 +25,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
28
+ diffusion_model.pt filter=lfs diff=lfs merge=lfs -text
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"_class_name": "UNetModel", "attn_resolutions": [16], "down_blocks": ["UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D"], "up_blocks": ["UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D"], "conv_resample": true, "downsample_padding": 0, "num_head_channels": null, "ch": 128, "ch_mult": [1, 2, 2, 2], "block_channels": [128, 256, 256, 256], "resnet_eps": 1e-06, "flip_sin_to_cos": false, "downscale_freq_shift": 1, "dropout": 0.1, "in_channels": 3, "out_channels": 3, "name_or_path": "./ddpm-cifar10/", "num_res_blocks": 2, "out_ch": 3, "resamp_with_conv": true, "resolution": 32, "image_size": 32, "ddpm": true}
diffusion_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:875d69676520406119756def41b3de3e1a034af762cff80c550fd74644e6f984
3
+ size 148401829