admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 09:46:00 +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
src/lerobot/policies/pi05_openpi/modeling_pi05openpi.py
+1 -1
View File
@@ -520,7 +520,7 @@ class PI05Pytorch(nn.Module): # see openpi `PI0Pytorch`
msg = """transformers_replace is not installed correctly. msg = """transformers_replace is not installed correctly.
Please install it with `pip install transformers==4.53.2` Please install it with `pip install transformers==4.53.2`
and `cp -r ./src/lerobot/policies/pi0_openpi/transformers_replace/* \ 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: try:
from transformers.models.siglip import check from transformers.models.siglip import check
src/lerobot/policies/pi0_openpi/modeling_pi0openpi.py
+1 -1
View File
@@ -521,7 +521,7 @@ class PI0Pytorch(nn.Module): # see openpi `PI0Pytorch`
msg = """transformers_replace is not installed correctly. msg = """transformers_replace is not installed correctly.
Please install it with `pip install transformers==4.53.2` Please install it with `pip install transformers==4.53.2`
and `cp -r ./src/lerobot/policies/pi0_openpi/transformers_replace/* \ 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: try:
from transformers.models.siglip import check from transformers.models.siglip import check