ISdept commited on
Commit
192d6ab
·
verified ·
1 Parent(s): 5074461

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. config.json +6 -3
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "type": "transformer_diffusion",
3
- "n_obs_steps": 4,
4
  "input_features": {
5
  "observation.state": {
6
  "type": "STATE",
@@ -64,12 +64,15 @@
64
  ],
65
  "num_cameras": 3,
66
  "vision_freeze_layers": 0,
 
 
 
67
  "state_dim": 7,
68
  "action_dim": 7,
69
  "d_model": 512,
70
  "nhead": 8,
71
- "num_encoder_layers": 3,
72
- "num_decoder_layers": 6,
73
  "dim_feedforward": 512,
74
  "diffusion_step_embed_dim": 256,
75
  "kernel_size": 3,
 
1
  {
2
  "type": "transformer_diffusion",
3
+ "n_obs_steps": 2,
4
  "input_features": {
5
  "observation.state": {
6
  "type": "STATE",
 
64
  ],
65
  "num_cameras": 3,
66
  "vision_freeze_layers": 0,
67
+ "cameras_for_vision_state_concat": [
68
+ "observation.images.front"
69
+ ],
70
  "state_dim": 7,
71
  "action_dim": 7,
72
  "d_model": 512,
73
  "nhead": 8,
74
+ "num_encoder_layers": 4,
75
+ "num_decoder_layers": 4,
76
  "dim_feedforward": 512,
77
  "diffusion_step_embed_dim": 256,
78
  "kernel_size": 3,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:63a9f053a6fef77ea9341eb5f9c8f474c8bdf8aa3ea628951f8a5ec476ef2d08
3
- size 1137820532
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c58b80bbf26a6470e5d2d3fc1682f0fae60ac6dc0bb3be3e3b7a5a2c49fc267
3
+ size 446938684