Views
No views yet
1from openpi.models_pytorch.pi0_pytorch import PI0Pytorch
2from safetensors.torch import load_file
3from huggingface_hub import hf_hub_download
4
5# Download model weights
6model_path = hf_hub_download(repo_id="your-repo-id", filename="model.safetensors")
7state_dict = load_file(model_path)
8
9# Load model (adjust config as needed)
10model = PI0Pytorch(...)
11model.load_state_dict(state_dict)