-
Notifications
You must be signed in to change notification settings - Fork 938
add LeRobot ACT notebook #3111
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: latest
Are you sure you want to change the base?
add LeRobot ACT notebook #3111
Conversation
|
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
| @@ -0,0 +1,598 @@ | |||
| { | |||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Line #1. # Export ACT Model to ONNX
Can we skip this step, and convert it to OV IR directly ?https://docs.openvino.ai/2025/openvino-workflow/model-preparation/convert-model-pytorch.html
Reply via ReviewNB
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Added another section to also convert it to OV IR directly.
| @@ -0,0 +1,598 @@ | |||
| { | |||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think we need an example to teach developer how to deploy this IR with OpenVINO
Reply via ReviewNB
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Added evaluation section which shows how to run the model.
|
|
||
| ENV_NAME="unitree_lerobot" | ||
| LEROBOT_DIR="unitree_IL_lerobot" | ||
| PY_VERSION="3.10" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why should we limit the version of python and force developer to use Conda.
For user experience, a notebook should be run across different OS and python virtual environment.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It's due to LeRobot requirements.
| python -m pip install --upgrade pip wheel setuptools | ||
| } | ||
|
|
||
| install_openvino_requirements() { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
please, add link to the quide https://github.com/openvinotoolkit/openvino_notebooks/blob/latest/README.md#-installation-guide to the README instead of processing it here
| @@ -0,0 +1,598 @@ | |||
| { | |||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Line #38. def pip_install(*packages):
we have such function in utils, please, check example in Prerequisites here https://github.com/openvinotoolkit/openvino_notebooks/blob/latest/notebooks/llm-native-agent-react/llm-native-agent-react.ipynb and use it
Reply via ReviewNB
| @@ -0,0 +1,598 @@ | |||
| { | |||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Where model.safetensors are originally placed ? On which step and from where should we download it ?
Reply via ReviewNB
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We need to create a download link for our model or user can use their own model.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could you, please, share some models so I can run it on my machine ?
| @@ -0,0 +1,598 @@ | |||
| { | |||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| @@ -0,0 +1,598 @@ | |||
| { | |||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
mo is deprecated, could you please convert model via ov.convert_model(model) https://docs.openvino.ai/2025/openvino-workflow/model-preparation/convert-model-pytorch.html ?
Reply via ReviewNB
openvino-dev-samples
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi @mhpanah
I can not install the dependencies on my ubuntu system properly.
Have you verified everything from scratch on your side.
Please run this in your terminal before running the notebook: bash setup_unitree_lerobot_env.sh
conda create -n unitree_lerobot python=3.10 -y
conda activate unitree_lerobot
# Launch notebook with correct kernel
jupyter lab act_to_openvino.ipynb --NotebookApp.kernel_name=unitree_lerobotThis is in the Minimal Installation & Launch section of the readme. |
Thanks for clarification. However we can not limit using Conda only, and any specific dependencies of a notebook should be included in the |

This PR adds a new tutorial notebook act_to_openvino.ipynb that demonstrates converting a Unitree Action Chunking Transformer (ACT) policy (from the LeRobot format) into OpenVINO IR (FP32), INT8. It walks through dependency verification, checkpoint preparation, ONNX export, OpenVINO Model Optimizer invocation, artifact validation, INT8 quantization and Evaluation.
Conversion Flow
Core Dependencies: