build in name scope

huggingface · Dec 7, 2023 · a130cbc · a130cbc
1 parent 25550fb
commit a130cbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/opt/test_modeling_tf_opt.py b/tests/models/opt/test_modeling_tf_opt.py
@@ -180,7 +180,7 @@ def _get_word_embedding_weight(model, embedding_layer):
             else:
                 # Here we build the word embeddings weights if not exists.
                 # And then we retry to get the attribute once built.
-                model.build()
+                model.build_in_name_scope()
                 if hasattr(embedding_layer, "weight"):
                     return embedding_layer.weight
                 else: