ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-03 20:31:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(modeling pi0): nit warning message

Browse Source
This commit is contained in:
Pepijn
2025-09-12 20:06:06 +02:00
parent e94844fa59
commit f840d2e006
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/lerobot/policies/pi05_openpi/modeling_pi05openpi.py
View File

@@ -520,7 +520,7 @@ class PI05Pytorch(nn.Module): # see openpi `PI0Pytorch`
msg = """transformers_replace is not installed correctly.
Please install it with `pip install transformers==4.53.2`
and `cp -r ./src/lerobot/policies/pi0_openpi/transformers_replace/* \
$(python -c "import transformers, os; print(os.path.dirname(transformers.__file__))")"""
$(python -c "import transformers, os; print(os.path.dirname(transformers.__file__))")`"""
try:
from transformers.models.siglip import check

2
src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py
View File

@@ -521,7 +521,7 @@ class PI0Pytorch(nn.Module): # see openpi `PI0Pytorch`
msg = """transformers_replace is not installed correctly.
Please install it with `pip install transformers==4.53.2`
and `cp -r ./src/lerobot/policies/pi0_openpi/transformers_replace/* \
$(python -c "import transformers, os; print(os.path.dirname(transformers.__file__))")"""
$(python -c "import transformers, os; print(os.path.dirname(transformers.__file__))")`"""
try:
from transformers.models.siglip import check