More TF fixes (#28081)

* More build_in_name_scope()

* Make sure we set the save spec now we don't do it with dummies anymore

* make fixup
This commit is contained in:
Matt
2023-12-18 15:26:03 +00:00
committed by GitHub
parent 0695b2421a
commit 71d47f0ad4
5 changed files with 6 additions and 5 deletions

View File

@@ -211,7 +211,7 @@ class TFAutoModelTest(unittest.TestCase):
config = copy.deepcopy(model.config)
config.architectures = ["FunnelBaseModel"]
model = TFAutoModel.from_config(config)
model.build()
model.build_in_name_scope()
self.assertIsInstance(model, TFFunnelBaseModel)
@@ -249,7 +249,7 @@ class TFAutoModelTest(unittest.TestCase):
config = NewModelConfig(**tiny_config.to_dict())
model = auto_class.from_config(config)
model.build()
model.build_in_name_scope()
self.assertIsInstance(model, TFNewModel)