admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

add changes

Browse Source
This commit is contained in:
jade.choghari@huggingface.co
2025-11-10 14:53:17 +00:00
parent 8d9a992953
commit 2219c29690
7 changed files with 130 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_xvla.py
+1 -1
View File
@@ -16,7 +16,7 @@ for name, param in policy.state_dict().items():
import safetensors.torch
from huggingface_hub import snapshot_download
cache_dir = snapshot_download(repo_id="2toINF/X-VLA-Libero", repo_type="model")
cache_dir = snapshot_download(repo_id="2toINF/X-VLA-Libero", repo_type="model", cache_dir="/fsx/jade_choghari/.cache/huggingface/model")
state_dict = safetensors.torch.load_file(os.path.join(cache_dir, "model.safetensors"))
# policy.load_state_dict(state_dict)
# 3. Add "model." prefix to every key