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
f321889
·
verified ·
1 Parent(s): a4dd75a

Upload folder using huggingface_hub

Browse files
Files changed (14) hide show
  1. checkpoint-100000/README.md +9 -0
  2. checkpoint-100000/config.json +49 -0
  3. checkpoint-100000/model.safetensors +3 -0
  4. checkpoint-100000/optimizer.bin +3 -0
  5. checkpoint-100000/pytorch_model.bin +3 -0
  6. checkpoint-100000/random_states_0.pkl +3 -0
  7. checkpoint-100000/random_states_1.pkl +3 -0
  8. checkpoint-100000/random_states_2.pkl +3 -0
  9. checkpoint-100000/random_states_3.pkl +3 -0
  10. checkpoint-100000/random_states_4.pkl +3 -0
  11. checkpoint-100000/random_states_5.pkl +3 -0
  12. checkpoint-100000/random_states_6.pkl +3 -0
  13. checkpoint-100000/random_states_7.pkl +3 -0
  14. checkpoint-100000/scheduler.bin +3 -0
checkpoint-100000/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-100000/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-100000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8afdc34710a5ade0380bc1b422a15e7adf6fd56d4823247b5f5aa3e0c299e7ef
3
+ size 2512832968
checkpoint-100000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb72982dec4bfb97707faf6c592259a331325a1244b53e2fc793c743f7268988
3
+ size 5026055309
checkpoint-100000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18b553fa786acd8b2471278128363caec23b875b0effc03fd714684fc853163b
3
+ size 2513008267
checkpoint-100000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44a3d7d9a9e1d6cde9f02dbab997074ef326e6423c9367af931663b4a800192e
3
+ size 16513
checkpoint-100000/random_states_1.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:724a82d452000510dfb5d25cace31610bda7f0dcdbdaed6f804c68b69738a57f
3
+ size 16449
checkpoint-100000/random_states_2.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66b9c87ced1bc9bb3274778805e537a47087e00969e687545b9162f9583f14d5
3
+ size 16513
checkpoint-100000/random_states_3.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c062568e030f6860cf77a08f1ed9cce91a97892fff9734e92f0cfcb24628300
3
+ size 16577
checkpoint-100000/random_states_4.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e8b2e9bf09fb21ed31a609b26ded5a7c0b6eae56c6a2792fa0afbe4193b4566
3
+ size 16513
checkpoint-100000/random_states_5.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f94dc714c03deefa0d0b21324c1d4096afd6d83fb1bcf49de4d87ddc26d84c0
3
+ size 16513
checkpoint-100000/random_states_6.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aaef5d44daa0f860a9d29f33d8d6231f519e85bf313e4e3fe3a846998850dff0
3
+ size 16513
checkpoint-100000/random_states_7.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6cdf908e235f91ade5486e34e5c179c81da1c6b75f1a8a24bd89ca455d403678
3
+ size 16513
checkpoint-100000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d7fa4eab419b19a5936e1f51b836bd2a8e02ffdcfa49e5fd06aa42b6ee08ca5
3
+ size 1465