hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests 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
import torch
from pytest import Cache
from torch import nn
from transformers import (
AutoConfig,
@@ -409,7 +408,7 @@ class SmolVLMWithExpertModel(nn.Module):
self,
attention_mask: Optional[torch.Tensor] = 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,
use_cache: Optional[bool] = None,
fill_kv_cache: Optional[bool] = None,