robotics-diffusion-transformer commited on
Commit
a6fe3f3
·
verified ·
1 Parent(s): 1759d4e

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. config.json +53 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "act_adaptor": "mlp3x_silu",
3
+ "act_pos_emb_config": [
4
+ [
5
+ "action",
6
+ 24
7
+ ],
8
+ [
9
+ "register",
10
+ 4
11
+ ]
12
+ ],
13
+ "action_dim": 20,
14
+ "img_pos_emb_config": null,
15
+ "img_token_dim": 2176,
16
+ "lang_pos_emb_config": null,
17
+ "lang_token_dim": null,
18
+ "max_img_len": null,
19
+ "max_lang_len": null,
20
+ "noise_scheduler": {
21
+ "num_inference_timesteps": 5
22
+ },
23
+ "pred_horizon": 24,
24
+ "rdt": {
25
+ "depth": 14,
26
+ "ffn_dim_multiplier": null,
27
+ "hidden_size": 1024,
28
+ "multiple_of": 256,
29
+ "norm_eps": 1e-05,
30
+ "num_heads": 8,
31
+ "num_kv_heads": 4,
32
+ "num_register_tokens": 4,
33
+ "use_flash_attn": true
34
+ },
35
+ "selected_layers": [
36
+ 0,
37
+ 1,
38
+ 2,
39
+ 3,
40
+ 4,
41
+ 5,
42
+ 6,
43
+ 7,
44
+ 8,
45
+ 9,
46
+ 10,
47
+ 11,
48
+ 12,
49
+ 13
50
+ ],
51
+ "state_adaptor": "mlp3x_silu",
52
+ "state_dim": 20
53
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fbdbce79fc2448432093cab2215b4f3e4e68bca0807d86f212631fbbba38106c
3
+ size 975438675