riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 05:10:06 +06:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Improve test_initialization for SwiftFormer (#38636)

Browse Source 
* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2025-06-06 10:47:10 +02:00 committed by GitHub
parent 92a87134ea
commit fca6748246
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/swiftformer/test_modeling_swiftformer.py
View File

@ -250,6 +250,8 @@ class SwiftFormerModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestC
for model_class in self.all_model_classes: for model_class in self.all_model_classes:
model = model_class(config=configs_no_init) model = model_class(config=configs_no_init)
for name, param in model.named_parameters(): for name, param in model.named_parameters():
if name.endswith(".w_g"):
continue
if param.requires_grad: if param.requires_grad:
self.assertIn( self.assertIn(
((param.data.mean() * 1e9) / 1e9).round().item(), ((param.data.mean() * 1e9) / 1e9).round().item(),