Improve test_initialization for SwiftFormer (#38636)

* 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

View File

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