Nathan Lambert commited on
Commit
ff13ce2
1 Parent(s): df16418

update mid and out block

Browse files
Files changed (2) hide show
  1. config.json +4 -1
  2. diffusion_pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,8 +1,9 @@
1
  {
2
  "_class_name": "UNet1DModel",
3
  "_diffusers_version": "0.5.0.dev0",
4
- "_name_or_path": "/users/nato/Downloads",
5
  "act_fn": "mish",
 
6
  "block_out_channels": [
7
  32,
8
  128,
@@ -15,7 +16,9 @@
15
  ],
16
  "in_channels": 14,
17
  "layers_per_block": 1,
 
18
  "norm_num_groups": 8,
 
19
  "out_channels": 14,
20
  "up_block_types": [
21
  "UpResnetBlock1D",
1
  {
2
  "_class_name": "UNet1DModel",
3
  "_diffusers_version": "0.5.0.dev0",
4
+ "_name_or_path": "fusing/ddpm-unet-rl-hopper-hor128",
5
  "act_fn": "mish",
6
+ "always_downsample": false,
7
  "block_out_channels": [
8
  32,
9
  128,
16
  ],
17
  "in_channels": 14,
18
  "layers_per_block": 1,
19
+ "mid_block_type": "MidResTemporalBlock1D",
20
  "norm_num_groups": 8,
21
+ "out_block_type": "OutConv1DBlock",
22
  "out_channels": 14,
23
  "up_block_types": [
24
  "UpResnetBlock1D",
diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:860797ea3dce1d1bc4d1617a1227a3c27641324b42560fec20102000850b032b
3
- size 14791449
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ee9974c82f783be1480050b3e68b32d966391578cc27a5aa820b6339dc86464
3
+ size 14792025