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

Replace build() with build_in_name_scope() for some TF tests (#28046)

Browse Source 
Replace build() with build_in_name_scope() for some tests
This commit is contained in:
Matt
2023-12-14 17:42:25 +00:00
committed by GitHub
parent 050e0b44f6
commit 3060899be5
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/ctrl/test_modeling_tf_ctrl.py
View File

@@ -225,7 +225,7 @@ class TFCTRLModelTest(TFModelTesterMixin, PipelineTesterMixin, unittest.TestCase
for model_class in self.all_model_classes:
model = model_class(config)
model.build() # may be needed for the get_bias() call below
model.build_in_name_scope() # may be needed for the get_bias() call below
assert isinstance(model.get_input_embeddings(), tf.keras.layers.Layer)
if model_class in list_lm_models: