albus2024 commited on
Commit
68a9edd
·
verified ·
1 Parent(s): 4fa5f72

Synchronize Dita libero_10_no_noops LeRobot checkpoint

Browse files
README.md CHANGED
@@ -7,7 +7,7 @@ tags:
7
  - libero
8
  ---
9
 
10
- # Legacy Dita — libero_10_no_noops
11
 
12
  This repository is a strict LeRobot conversion of `libero_long.pth` from the original
13
  [Dita](https://github.com/RoboDita/Dita) implementation. It is configured for the
@@ -17,7 +17,7 @@ This repository is a strict LeRobot conversion of `libero_long.pth` from the ori
17
 
18
  - Source checkpoint: `/home/lyanam/project/lerobot_policy_dita/model_ckpt/libero_long.pth`
19
  - Source checkpoint SHA256: `4342de0cfaccc335b6288e378038db2748a703f6187217f50c340f37d49ffa7a`
20
- - Conversion code commit: `83fbfda3dfc09d5cae6d59bdc80aa4def39ec98a`
21
  - Dataset statistics: `/home/shared/lyanam/huggingface/hub/datasets--openvla--modified_libero_rlds/snapshots/6ce6aaaaabdbe590b1eef5cd29c0d33f14a08551/libero_10_no_noops/1.0.0/dataset_statistics_3f36439171368ac499e64c677362e590ff6e0b2a87996780fc0e486ac04fd7a4.json`
22
  - Dataset statistics SHA256: `e8b8c1dd10b983643a4aeb532b5e245437565bd12c66dbce431c59dda903c1ef`
23
  - Image processing: one RGB primary/agentview camera, resized to 224×224, followed once by
@@ -27,7 +27,7 @@ This repository is a strict LeRobot conversion of `libero_long.pth` from the ori
27
  - q01: `[-0.6348214149475098, -0.7741071581840515, -0.7633928656578064, -0.09749999642372131, -0.14819999992847435, -0.2742857038974762, 0.0]`
28
  - q99: `[0.7714285850524902, 0.8464285731315613, 0.9375, 0.13928571343421936, 0.15964286029338837, 0.3246428668498993, 1.0]`
29
  - Gripper: normalized output is thresholded with `> 0.5`, then the Cartesian dimensions are
30
- unnormalized, then the legacy gripper sign flip is applied.
31
  - Rollout: `traj_length=11`, two-frame history, `chunk_size=10`, `n_action_steps=1`,
32
  `num_inference_steps=10`, DDIM `prediction_type=epsilon`, and CUDA AMP enabled.
33
  - LIBERO's LeRobot environment processor performs the 180° image rotation. The policy does not
@@ -36,7 +36,7 @@ This repository is a strict LeRobot conversion of `libero_long.pth` from the ori
36
  ## Install
37
 
38
  ```bash
39
- pip install "git+https://github.com/A-Dumbledore/lerobot_policy_dita.git@83fbfda3dfc09d5cae6d59bdc80aa4def39ec98a"
40
  ```
41
 
42
  ## Evaluate
@@ -58,6 +58,6 @@ local dataset metadata is needed at evaluation time.
58
 
59
  ## Validation
60
 
61
- The conversion was gated on strict state-dict loading, old/new core-forward parity, ImageNet image
62
  parity, CLIP token/mask/embedding parity, online history tests, suite-specific masked action parity,
63
  10-step DDIM parity, local serialization/reload, and a clean Hugging Face download smoke test.
 
7
  - libero
8
  ---
9
 
10
+ # Dita — libero_10_no_noops
11
 
12
  This repository is a strict LeRobot conversion of `libero_long.pth` from the original
13
  [Dita](https://github.com/RoboDita/Dita) implementation. It is configured for the
 
17
 
18
  - Source checkpoint: `/home/lyanam/project/lerobot_policy_dita/model_ckpt/libero_long.pth`
19
  - Source checkpoint SHA256: `4342de0cfaccc335b6288e378038db2748a703f6187217f50c340f37d49ffa7a`
20
+ - Conversion code commit: `f506ceccb9b2816ffebb024b14aeaa2b34fbd46b`
21
  - Dataset statistics: `/home/shared/lyanam/huggingface/hub/datasets--openvla--modified_libero_rlds/snapshots/6ce6aaaaabdbe590b1eef5cd29c0d33f14a08551/libero_10_no_noops/1.0.0/dataset_statistics_3f36439171368ac499e64c677362e590ff6e0b2a87996780fc0e486ac04fd7a4.json`
22
  - Dataset statistics SHA256: `e8b8c1dd10b983643a4aeb532b5e245437565bd12c66dbce431c59dda903c1ef`
23
  - Image processing: one RGB primary/agentview camera, resized to 224×224, followed once by
 
27
  - q01: `[-0.6348214149475098, -0.7741071581840515, -0.7633928656578064, -0.09749999642372131, -0.14819999992847435, -0.2742857038974762, 0.0]`
28
  - q99: `[0.7714285850524902, 0.8464285731315613, 0.9375, 0.13928571343421936, 0.15964286029338837, 0.3246428668498993, 1.0]`
29
  - Gripper: normalized output is thresholded with `> 0.5`, then the Cartesian dimensions are
30
+ unnormalized, then the original Dita gripper sign flip is applied.
31
  - Rollout: `traj_length=11`, two-frame history, `chunk_size=10`, `n_action_steps=1`,
32
  `num_inference_steps=10`, DDIM `prediction_type=epsilon`, and CUDA AMP enabled.
33
  - LIBERO's LeRobot environment processor performs the 180° image rotation. The policy does not
 
36
  ## Install
37
 
38
  ```bash
39
+ pip install "git+https://github.com/A-Dumbledore/lerobot_policy_dita.git@f506ceccb9b2816ffebb024b14aeaa2b34fbd46b"
40
  ```
41
 
42
  ## Evaluate
 
58
 
59
  ## Validation
60
 
61
+ The conversion was gated on strict state-dict loading, source/plugin core-forward parity, ImageNet image
62
  parity, CLIP token/mask/embedding parity, online history tests, suite-specific masked action parity,
63
  10-step DDIM parity, local serialization/reload, and a clean Hugging Face download smoke test.
config.json CHANGED
@@ -28,8 +28,7 @@
28
  "tags": null,
29
  "license": null,
30
  "pretrained_path": null,
31
- "compatibility_mode": "legacy_dita",
32
- "legacy_suite": "libero_10_no_noops",
33
  "num_layers": 12,
34
  "traj_length": 11,
35
  "dropout_rate": 0.0,
@@ -45,7 +44,6 @@
45
  ],
46
  "chunk_size": 10,
47
  "n_action_steps": 1,
48
- "image_preprocessing": "imagenet",
49
  "history_frames": 2,
50
  "gripper_threshold": 0.5,
51
  "gripper_sign_flip": -1.0,
 
28
  "tags": null,
29
  "license": null,
30
  "pretrained_path": null,
31
+ "dataset_suite": "libero_10_no_noops",
 
32
  "num_layers": 12,
33
  "traj_length": 11,
34
  "dropout_rate": 0.0,
 
44
  ],
45
  "chunk_size": 10,
46
  "n_action_steps": 1,
 
47
  "history_frames": 2,
48
  "gripper_threshold": 0.5,
49
  "gripper_sign_flip": -1.0,
conversion_metadata.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "converted_at_utc": "2026-08-04T05:34:43.528681+00:00",
3
  "source_model": "Dita/scripts/llama_dp.py",
4
  "source_checkpoint": "model_ckpt/libero_long.pth",
5
  "source_checkpoint_sha256": "4342de0cfaccc335b6288e378038db2748a703f6187217f50c340f37d49ffa7a",
@@ -10,7 +10,7 @@
10
  "unexpected_keys": [],
11
  "shape_mismatches": []
12
  },
13
- "legacy_suite": "libero_10_no_noops",
14
  "dataset_statistics_file": "/home/shared/lyanam/huggingface/hub/datasets--openvla--modified_libero_rlds/snapshots/6ce6aaaaabdbe590b1eef5cd29c0d33f14a08551/libero_10_no_noops/1.0.0/dataset_statistics_3f36439171368ac499e64c677362e590ff6e0b2a87996780fc0e486ac04fd7a4.json",
15
  "dataset_statistics_sha256": "e8b8c1dd10b983643a4aeb532b5e245437565bd12c66dbce431c59dda903c1ef",
16
  "action_q01": [
@@ -40,10 +40,11 @@
40
  true,
41
  false
42
  ],
43
- "conversion_code_git_sha": "83fbfda3dfc09d5cae6d59bdc80aa4def39ec98a",
44
  "target_repo_id": "albus2024/dita_long",
45
  "processor_state_files": [
46
  "policy_postprocessor_step_0_dita_masked_unnormalizer_processor.safetensors",
47
  "policy_preprocessor_step_4_dita_masked_normalizer_processor.safetensors"
48
- ]
 
49
  }
 
1
  {
2
+ "converted_at_utc": "2026-08-04T06:20:04.205869+00:00",
3
  "source_model": "Dita/scripts/llama_dp.py",
4
  "source_checkpoint": "model_ckpt/libero_long.pth",
5
  "source_checkpoint_sha256": "4342de0cfaccc335b6288e378038db2748a703f6187217f50c340f37d49ffa7a",
 
10
  "unexpected_keys": [],
11
  "shape_mismatches": []
12
  },
13
+ "dataset_suite": "libero_10_no_noops",
14
  "dataset_statistics_file": "/home/shared/lyanam/huggingface/hub/datasets--openvla--modified_libero_rlds/snapshots/6ce6aaaaabdbe590b1eef5cd29c0d33f14a08551/libero_10_no_noops/1.0.0/dataset_statistics_3f36439171368ac499e64c677362e590ff6e0b2a87996780fc0e486ac04fd7a4.json",
15
  "dataset_statistics_sha256": "e8b8c1dd10b983643a4aeb532b5e245437565bd12c66dbce431c59dda903c1ef",
16
  "action_q01": [
 
40
  true,
41
  false
42
  ],
43
+ "conversion_code_git_sha": "f506ceccb9b2816ffebb024b14aeaa2b34fbd46b",
44
  "target_repo_id": "albus2024/dita_long",
45
  "processor_state_files": [
46
  "policy_postprocessor_step_0_dita_masked_unnormalizer_processor.safetensors",
47
  "policy_preprocessor_step_4_dita_masked_normalizer_processor.safetensors"
48
+ ],
49
+ "previous_hub_revision": "4fa5f721f7f16621361a8b23fa77cee1bc1c952f"
50
  }
policy_preprocessor.json CHANGED
@@ -28,8 +28,7 @@
28
  "clip_model_name": "openai/clip-vit-large-patch14",
29
  "traj_length": 11,
30
  "chunk_size": 10,
31
- "device": "cuda",
32
- "use_attention_mask": true
33
  }
34
  },
35
  {
 
28
  "clip_model_name": "openai/clip-vit-large-patch14",
29
  "traj_length": 11,
30
  "chunk_size": 10,
31
+ "device": "cuda"
 
32
  }
33
  },
34
  {