Add diffusion model
Browse files- .gitattributes +1 -0
- config.json +1 -0
- 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
|