Nathan Lambert commited on
Commit
205dbd4
1 Parent(s): 7663990

pretrained works again

Browse files
Files changed (2) hide show
  1. config.json +8 -2
  2. diffusion_pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,6 +1,12 @@
1
  {
2
  "_class_name": "UNet1DModel",
3
- "_diffusers_version": "0.0.4",
 
4
  "dim": 32,
5
- "dim_mults": [1, 4, 8]
 
 
 
 
 
6
  }
 
1
  {
2
  "_class_name": "UNet1DModel",
3
+ "_diffusers_version": "0.5.0.dev0",
4
+ "_name_or_path": "fusing/ddpm-unet-rl-hopper-hor128",
5
  "dim": 32,
6
+ "dim_mults": [
7
+ 1,
8
+ 4,
9
+ 8
10
+ ],
11
+ "transition_dim": 14
12
  }
diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:26ce44a2d2585e92755431b9b6d8dce011f2332fc763d45837ab8986b32ab032
3
- size 14792345
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61ed95d12dd872cb83c1aba632866fd09028fbbaff62d55d68dc140a4656ea63
3
+ size 14790745