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

oops, typo

Browse Source
This commit is contained in:
raushan
2026-04-07 12:40:42 +02:00
committed by Steven Palma
parent f52b4a1de3
commit 430a2e3f2b
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi0/modeling_pi0.py
+1 -1
View File
@@ -450,7 +450,7 @@ class PaliGemmaWithExpertModel(
return features return features
def embed_language_tokens(self, tokens: torch.Tensor): def embed_language_tokens(self, tokens: torch.Tensor):
return self.paligemma.model.language_model.get_inputs_embeddings()(tokens) return self.paligemma.model.language_model.get_input_embeddings()(tokens)
def forward( def forward(
self, self,
src/lerobot/policies/pi0_fast/modeling_pi0_fast.py
+1 -1
View File
@@ -266,7 +266,7 @@ class PI0FastPaliGemma(nn.Module):
return features return features
def embed_language_tokens(self, tokens: torch.Tensor): def embed_language_tokens(self, tokens: torch.Tensor):
return self.paligemma.model.language_model.get_inputs_embeddings()(tokens) return self.paligemma.model.language_model.get_input_embeddings()(tokens)
def forward( def forward(
self, self,