Views
No views yet
1from gr00t.model.policy import Gr00tPolicy
2from gr00t.data.embodiment_tags import EmbodimentTag
3
4# Load the model
5policy = Gr00tPolicy(
6 model_path="namaewa-im/spot-gr00t-task0",
7 embodiment_tag=EmbodimentTag.NEW_EMBODIMENT,
8 device="cuda"
9)
10
11# Use for inference
12actions = policy.get_action(input_data)