Hugging Face's logo

dflorea
/
checkpoints-1b-single

Safetensors
Model card Files
xet
Community
dflorea commited on
Commit
c166d8e
·
verified ·
1 Parent(s): 4c78c0d

Add files using upload-large-folder tool

Browse files
Files changed (16) hide show
  1. rdt-finetune-1b-maniskill-single/checkpoint-10000/README.md +9 -0
  2. rdt-finetune-1b-maniskill-single/checkpoint-10000/config.json +49 -0
  3. rdt-finetune-1b-maniskill-single/checkpoint-10000/ema/model.safetensors +3 -0
  4. rdt-finetune-1b-maniskill-single/checkpoint-10000/model.safetensors +3 -0
  5. rdt-finetune-1b-maniskill-single/checkpoint-10000/optimizer.bin +3 -0
  6. rdt-finetune-1b-maniskill-single/checkpoint-10000/pytorch_model.bin +3 -0
  7. rdt-finetune-1b-maniskill-single/checkpoint-10000/random_states_0.pkl +3 -0
  8. rdt-finetune-1b-maniskill-single/checkpoint-10000/scheduler.bin +3 -0
  9. rdt-finetune-1b-maniskill-single/checkpoint-20000/README.md +9 -0
  10. rdt-finetune-1b-maniskill-single/checkpoint-20000/config.json +49 -0
  11. rdt-finetune-1b-maniskill-single/checkpoint-20000/ema/model.safetensors +3 -0
  12. rdt-finetune-1b-maniskill-single/checkpoint-20000/model.safetensors +3 -0
  13. rdt-finetune-1b-maniskill-single/checkpoint-20000/optimizer.bin +3 -0
  14. rdt-finetune-1b-maniskill-single/checkpoint-20000/pytorch_model.bin +3 -0
  15. rdt-finetune-1b-maniskill-single/checkpoint-20000/random_states_0.pkl +3 -0
  16. rdt-finetune-1b-maniskill-single/checkpoint-20000/scheduler.bin +3 -0
rdt-finetune-1b-maniskill-single/checkpoint-10000/README.md ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - model_hub_mixin
4
+ - pytorch_model_hub_mixin
5
+ ---
6
+
7
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
+ - Library: https://huggingface.co/robotics-diffusion-transformer/rdt-1b
9
+ - Docs: [More Information Needed]
rdt-finetune-1b-maniskill-single/checkpoint-10000/config.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action_dim": 128,
3
+ "ema": {
4
+ "inv_gamma": 1.0,
5
+ "max_value": 0.9999,
6
+ "min_value": 0.0,
7
+ "power": 0.75,
8
+ "update_after_step": 0
9
+ },
10
+ "img_adaptor": "mlp2x_gelu",
11
+ "img_cond_len": 4374,
12
+ "img_pos_embed_config": [
13
+ [
14
+ "image",
15
+ [
16
+ 2,
17
+ 3,
18
+ -729
19
+ ]
20
+ ]
21
+ ],
22
+ "img_token_dim": 1152,
23
+ "lang_adaptor": "mlp2x_gelu",
24
+ "lang_pos_embed_config": [
25
+ [
26
+ "lang",
27
+ -1024
28
+ ]
29
+ ],
30
+ "lang_token_dim": 4096,
31
+ "max_lang_cond_len": 1024,
32
+ "noise_scheduler": {
33
+ "beta_schedule": "squaredcos_cap_v2",
34
+ "clip_sample": false,
35
+ "num_inference_timesteps": 5,
36
+ "num_train_timesteps": 1000,
37
+ "prediction_type": "sample",
38
+ "type": "ddpm"
39
+ },
40
+ "pred_horizon": 64,
41
+ "rdt": {
42
+ "cond_pos_embed_type": "multimodal",
43
+ "depth": 28,
44
+ "hidden_size": 2048,
45
+ "num_heads": 32
46
+ },
47
+ "state_adaptor": "mlp3x_gelu",
48
+ "state_token_dim": 128
49
+ }
rdt-finetune-1b-maniskill-single/checkpoint-10000/ema/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7dceefce3a2cc5619dc240a95096ba38f1957b7bc74562cb91a414ac6217c5b6
3
+ size 2456705456
rdt-finetune-1b-maniskill-single/checkpoint-10000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f43bc5dd7dd5604d8fe44fbe9c2dde1e80d6643e509a8c85a428d6dabc124345
3
+ size 2512832968
rdt-finetune-1b-maniskill-single/checkpoint-10000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5bdbb5527b8dd1e455e0df2f4e067c354bc7bb8aa3455047ecf67e259e76930f
3
+ size 5026055309
rdt-finetune-1b-maniskill-single/checkpoint-10000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:851c1a7fed52f75f62f4076b591145f0408fa4a53d345141bf948b656d6401f1
3
+ size 2513008267
rdt-finetune-1b-maniskill-single/checkpoint-10000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c533f544e5bc8b99e92436ccc0fa909d71b4ae288c6c984ff33c34a47466b770
3
+ size 14757
rdt-finetune-1b-maniskill-single/checkpoint-10000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ea39d5520f32c49588c091b92e3f737396699c2fa98cf95f99ff6f3f1f88983
3
+ size 1401
rdt-finetune-1b-maniskill-single/checkpoint-20000/README.md ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - model_hub_mixin
4
+ - pytorch_model_hub_mixin
5
+ ---
6
+
7
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
+ - Library: https://huggingface.co/robotics-diffusion-transformer/rdt-1b
9
+ - Docs: [More Information Needed]
rdt-finetune-1b-maniskill-single/checkpoint-20000/config.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action_dim": 128,
3
+ "ema": {
4
+ "inv_gamma": 1.0,
5
+ "max_value": 0.9999,
6
+ "min_value": 0.0,
7
+ "power": 0.75,
8
+ "update_after_step": 0
9
+ },
10
+ "img_adaptor": "mlp2x_gelu",
11
+ "img_cond_len": 4374,
12
+ "img_pos_embed_config": [
13
+ [
14
+ "image",
15
+ [
16
+ 2,
17
+ 3,
18
+ -729
19
+ ]
20
+ ]
21
+ ],
22
+ "img_token_dim": 1152,
23
+ "lang_adaptor": "mlp2x_gelu",
24
+ "lang_pos_embed_config": [
25
+ [
26
+ "lang",
27
+ -1024
28
+ ]
29
+ ],
30
+ "lang_token_dim": 4096,
31
+ "max_lang_cond_len": 1024,
32
+ "noise_scheduler": {
33
+ "beta_schedule": "squaredcos_cap_v2",
34
+ "clip_sample": false,
35
+ "num_inference_timesteps": 5,
36
+ "num_train_timesteps": 1000,
37
+ "prediction_type": "sample",
38
+ "type": "ddpm"
39
+ },
40
+ "pred_horizon": 64,
41
+ "rdt": {
42
+ "cond_pos_embed_type": "multimodal",
43
+ "depth": 28,
44
+ "hidden_size": 2048,
45
+ "num_heads": 32
46
+ },
47
+ "state_adaptor": "mlp3x_gelu",
48
+ "state_token_dim": 128
49
+ }
rdt-finetune-1b-maniskill-single/checkpoint-20000/ema/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0229b68346a2e58a52dca0541c034e0013d3861fcccc67d7f9cdf96440707c8f
3
+ size 2456705456
rdt-finetune-1b-maniskill-single/checkpoint-20000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b97eaa9148d97234df49aadc41bbca38a5aad9ad1143bc39840d2d803638391
3
+ size 2512832968
rdt-finetune-1b-maniskill-single/checkpoint-20000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16ae8196b56cfaaf9192c782cff109974ca150fa13b7432afd9a3ef0bc5d9f8c
3
+ size 5026055309
rdt-finetune-1b-maniskill-single/checkpoint-20000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3416764677e9cd699106f862f3115ae2eb4c8b25a2a852e622c3d6a445eea08
3
+ size 2513008267
rdt-finetune-1b-maniskill-single/checkpoint-20000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76dc317d8a27a9658627a5b199cf472c2b34f3ee293888dd77f5fc1880322c53
3
+ size 14757
rdt-finetune-1b-maniskill-single/checkpoint-20000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:25731918d1f99c47a3422243d7804801967b18607df0e5678917520261d87cab
3
+ size 1401