tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

avoid import pytest

Browse Source
This commit is contained in:
mshukor
2025-06-01 09:36:23 +02:00
parent 4027d97c70
commit 5fd40b188e
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/common/policies/smolvla/smolvlm_with_expert.py
View File

@@ -16,7 +16,6 @@ import copy
from typing import List, Optional, Union from typing import List, Optional, Union
import torch import torch
from pytest import Cache
from torch import nn from torch import nn
from transformers import ( from transformers import (
AutoConfig, AutoConfig,
@@ -409,7 +408,7 @@ class SmolVLMWithExpertModel(nn.Module):
self, self,
attention_mask: Optional[torch.Tensor] = None, attention_mask: Optional[torch.Tensor] = None,
position_ids: Optional[torch.LongTensor] = None, position_ids: Optional[torch.LongTensor] = None,
past_key_values: Optional[Union[List[torch.FloatTensor], Cache]] = None, past_key_values: Optional[List[torch.FloatTensor]] = None,
inputs_embeds: List[torch.FloatTensor] = None, inputs_embeds: List[torch.FloatTensor] = None,
use_cache: Optional[bool] = None, use_cache: Optional[bool] = None,
fill_kv_cache: Optional[bool] = None, fill_kv_cache: Optional[bool] = None,