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

4
examples/modular-transformers/modeling_dummy.py
View File

@@ -881,9 +881,7 @@ class DummyModel(DummyPreTrainedModel):
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
if (input_ids is None) ^ (inputs_embeds is not None):
raise ValueError(
"You must specify exactly one of input_ids or inputs_embeds"
)
raise ValueError("You must specify exactly one of input_ids or inputs_embeds")
if self.gradient_checkpointing and self.training and use_cache:
logger.warning_once(

4
examples/modular-transformers/modeling_my_new_model2.py
View File

@@ -758,9 +758,7 @@ class MyNewModel2Model(MyNewModel2PreTrainedModel):
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
if (input_ids is None) ^ (inputs_embeds is not None):
raise ValueError(
"You must specify exactly one of input_ids or inputs_embeds"
)
raise ValueError("You must specify exactly one of input_ids or inputs_embeds")
if self.gradient_checkpointing and self.training and use_cache:
logger.warning_once(