SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

CI: fix failures (#34371)

Browse Source 
fix
This commit is contained in:
Raushan Turganbay
2024-10-24 13:44:53 +02:00
committed by GitHub
parent f0b3ef9e2e
commit b29c24ff1e
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/models/video_llava/modeling_video_llava.py
View File

@@ -561,6 +561,7 @@ class VideoLlavaForConditionalGeneration(VideoLlavaPreTrainedModel, GenerationMi
) )
video_features = None video_features = None
num_frames = 0
if pixel_values_videos is not None: if pixel_values_videos is not None:
video_features, num_frames = self.get_video_features( video_features, num_frames = self.get_video_features(
pixel_values_videos=pixel_values_videos, vision_feature_layer=vision_feature_layer pixel_values_videos=pixel_values_videos, vision_feature_layer=vision_feature_layer

2
tests/models/instructblip/test_modeling_instructblip.py
View File

@@ -621,7 +621,7 @@ class InstructBlipModelIntegrationTest(unittest.TestCase):
logits = model(**inputs).logits logits = model(**inputs).logits
expected_slice = torch.tensor( expected_slice = torch.tensor(
[[-3.3926, -12.2969, 8.4922], [-5.0195, -11.9531, 8.1406], [-4.0039, -13.3594, 9.2578]], [[-3.3047, -12.0625, 8.4922], [-4.9258, -11.7578, 8.1406], [-3.9297, -13.5000, 9.2500]],
device=torch_device, device=torch_device,
) )