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

Paligemma: fix static cache test (#33941)

Browse Source 
* fix

* not flaky anymore + style
This commit is contained in:
Raushan Turganbay
2024-10-05 09:47:37 +02:00
committed by GitHub
parent 38f9f10dd9
commit 612065efeb
5 changed files with 8 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/paligemma/test_modeling_paligemma.py
View File

@@ -159,7 +159,8 @@ class PaliGemmaVisionText2TextModelTester:
config_and_inputs = self.prepare_config_and_inputs()
config, pixel_values = config_and_inputs
input_ids = ids_tensor([self.batch_size, self.seq_length], config.text_config.vocab_size - 1) + 1
attention_mask = input_ids.ne(1).to(torch_device)
attention_mask = input_ids.ne(self.pad_token_id).to(torch_device)
# set the 16 first tokens to be image, and ensure that no other tokens are image tokens
# do not change this unless you modified image size or patch size
input_ids[input_ids == config.image_token_index] = self.pad_token_id