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

adhere to python 3.11 syntax

Browse Source
This commit is contained in:
Pepijn
2025-09-10 20:20:31 +02:00
parent 8178a06b90
commit 8c5fe10d6c
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi0_openpi/transformers_replace/models/gemma/modeling_gemma.py
+3 -2
View File
@@ -355,8 +355,9 @@ class GemmaDecoderLayer(GradientCheckpointingLayer):
output_attentions: bool | None = False, output_attentions: bool | None = False,
use_cache: bool | None = False, use_cache: bool | None = False,
cache_position: torch.LongTensor | None = None, cache_position: torch.LongTensor | None = None,
position_embeddings: None position_embeddings: (
| (tuple[torch.Tensor, torch.Tensor]) = None, # necessary, but kept here for BC None | tuple[torch.Tensor, torch.Tensor]
) = None, # necessary, but kept here for BC
adarms_cond: torch.Tensor | None = None, adarms_cond: torch.Tensor | None = None,
**kwargs: Unpack[FlashAttentionKwargs], **kwargs: Unpack[FlashAttentionKwargs],
) -> tuple[torch.FloatTensor, tuple[torch.FloatTensor, torch.FloatTensor] | None]: ) -> tuple[torch.FloatTensor, tuple[torch.FloatTensor, torch.FloatTensor] | None]: