Hugging Face's logo

yuta0x89
/
sample-rdt

PyTorch
TensorBoard
Safetensors
model_hub_mixin
pytorch_model_hub_mixin
Model card Files
xet
Metrics Community
yuta0x89 commited on
Commit
6c496dd
·
verified ·
1 Parent(s): f321889

Upload folder using huggingface_hub

Browse files
Files changed (14) hide show
  1. checkpoint-10000/README.md +9 -0
  2. checkpoint-10000/config.json +49 -0
  3. checkpoint-10000/model.safetensors +3 -0
  4. checkpoint-10000/optimizer.bin +3 -0
  5. checkpoint-10000/pytorch_model.bin +3 -0
  6. checkpoint-10000/random_states_0.pkl +3 -0
  7. checkpoint-10000/random_states_1.pkl +3 -0
  8. checkpoint-10000/random_states_2.pkl +3 -0
  9. checkpoint-10000/random_states_3.pkl +3 -0
  10. checkpoint-10000/random_states_4.pkl +3 -0
  11. checkpoint-10000/random_states_5.pkl +3 -0
  12. checkpoint-10000/random_states_6.pkl +3 -0
  13. checkpoint-10000/random_states_7.pkl +3 -0
  14. checkpoint-10000/scheduler.bin +3 -0
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]
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
+ }
checkpoint-10000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5dfcda8f79f7eaa1ded1d356ff7af589424678971a322b57c39ba3a8db368df9
3
+ size 2512832968
checkpoint-10000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4848899ba8ad439d2ea1b9a4d5c12419b9fadffbbfa43a102ba019de22eab46f
3
+ size 5026055309
checkpoint-10000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:53145a7af6df1aad62896c4cfe813d298bfcf4b619e85471223c05f55aea66f8
3
+ size 2513008267
checkpoint-10000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:369513ab041672991be6f8f07041706419434192a5c7014ab28ba5fc6abaf45e
3
+ size 16513
checkpoint-10000/random_states_1.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1524eeeb392d8763166cc3779962d4e5b70da6d9ba07a756d94045d2f6e48cb2
3
+ size 16449
checkpoint-10000/random_states_2.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d9834dddb82075c7a4a05c4b8c3879a48fac2124e67c359d58e8d3d1e22c71e
3
+ size 16513
checkpoint-10000/random_states_3.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:355580bac2d4fb1bc5c9f44aa6cf54720b1bce8141d341125bc251e4b38f7fe4
3
+ size 16577
checkpoint-10000/random_states_4.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c2de0219e304d2450bffa1ae4c308faf82cc09bce72bc831e0f501f53ec47fa9
3
+ size 16513
checkpoint-10000/random_states_5.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81982e1fa703ae327c51e37cc4b92bf558129189b97d6b7d8451e3ba7e22b9ac
3
+ size 16513
checkpoint-10000/random_states_6.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fd821d2d7081b1cb3848f231236a4ca3d018dc2c0d24f759fede0e6a8e93a56
3
+ size 16513
checkpoint-10000/random_states_7.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fbcc02aa968d3486e2a525d091fa7fadc955563d1186294c59c4fb6568d53fad
3
+ size 16513
checkpoint-10000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce68756df31e6f41aeacb13e25f7365e07202127452dc14f05aca847f57a1d5b
3
+ size 1465