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

Fix important models CI (#39576)

Browse Source 
* relax test boundaries and fix from config

* eager is always supported.
This commit is contained in:
Pablo Montalvo
2025-07-23 16:24:29 +02:00
committed by GitHub
parent 0fe03afeb8
commit ea56eb6bed
4 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_modeling_common.py
View File

@@ -4107,7 +4107,9 @@ class ModelTesterMixin:
config, _ = self.model_tester.prepare_config_and_inputs_for_common()
cls = self._torch_compile_train_cls # e.g. LlamaFroCausalLM
model = cls(config, attn_implementation="flash_attention_2").to(device=torch_device, dtype=torch_dtype)
model = cls._from_config(config, attn_implementation="flash_attention_2").to(
device=torch_device, dtype=torch_dtype
)
inputs = {
"input_ids": torch.randint(low=1, high=model.config.vocab_size, size=(2, 10), device=torch_device),
@@ -4137,7 +4139,7 @@ class ModelTesterMixin:
}
for model_class in self.all_generative_model_classes:
if not getattr(model_class, support_flag[attn_implementation]):
if attn_implementation != "eager" and not getattr(model_class, support_flag[attn_implementation]):
self.skipTest(f"{model_class.__name__} does not support {attn_implementation}")
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()