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

Defaults to adamw_torch_fused for Pytorch>=2.8 (#37358)

Browse Source 
* Defaults to adamw_torch_fused for latest Pytorch

Signed-off-by: cyy <cyyever@outlook.com>

* Fix test

Signed-off-by: cyy <cyyever@outlook.com>

---------

Signed-off-by: cyy <cyyever@outlook.com>
This commit is contained in:
Yuanyuan Chen
2025-07-16 17:52:33 +08:00
committed by GitHub
parent 4524a68c66
commit ae4e306a40
3 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/git/test_modeling_git.py
View File

@@ -347,7 +347,8 @@ class GitModelTester:
num_return_sequences=2,
)
self.parent.assertEqual(generated_ids.shape, (self.batch_size * 2, 20))
self.parent.assertEqual(generated_ids.shape[0], self.batch_size * 2)
self.parent.assertTrue(generated_ids.shape[1] < 20)
def _test_batched_generate_captioning(self, config, input_ids, input_mask, pixel_values):
model = GitForCausalLM(config=config)