Thanks for this great contribution.
One small thing, in your example, you wrotepolicy = Pi0Policy.from_pretrained("lerobot/pi0")
However, the policy in the source code is defined as PI0Policy
. I was confused by the import error for a long while.
Thanks for this great contribution.
One small thing, in your example, you wrotepolicy = Pi0Policy.from_pretrained("lerobot/pi0")
However, the policy in the source code is defined as PI0Policy
. I was confused by the import error for a long while.