tf-models-nightly 2.18.0.dev20240903__py2.py3-none-any.whl → 2.18.0.dev20240904__py2.py3-none-any.whl

official/nlp/modeling/networks/mobile_bert_encoder.py

@@ -162,7 +162,10 @@ class MobileBERTEncoder(tf_keras.Model):
         pooled_output=first_token,
         encoder_outputs=all_layer_outputs,
         attention_scores=all_attention_scores)
+    super().__init__(
+        inputs=self.inputs, outputs=outputs, **kwargs)
     self._config = dict(
+        name=self.name,
         word_vocab_size=word_vocab_size,
         word_embed_size=word_embed_size,
         type_vocab_size=type_vocab_size,
@@ -182,11 +185,9 @@ class MobileBERTEncoder(tf_keras.Model):
         normalization_type=normalization_type,
         classifier_activation=classifier_activation,
         input_mask_dtype=input_mask_dtype,
+        **kwargs,
     )
 
-    super().__init__(
-        inputs=self.inputs, outputs=outputs, **kwargs)
-
   def get_config(self):
     return dict(self._config)
 

{tf_models_nightly-2.18.0.dev20240903.dist-info → tf_models_nightly-2.18.0.dev20240904.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tf-models-nightly
-Version: 2.18.0.dev20240903
+Version: 2.18.0.dev20240904
 Summary: TensorFlow Official Models
 Home-page: https://github.com/tensorflow/models
 Author: Google Inc.

{tf_models_nightly-2.18.0.dev20240903.dist-info → tf_models_nightly-2.18.0.dev20240904.dist-info}/RECORD

@@ -407,7 +407,7 @@ official/nlp/modeling/networks/fnet.py,sha256=DAIQCixcqDoZjGSEo1apm3pAFlm9Qlsn1I
 official/nlp/modeling/networks/fnet_test.py,sha256=mCQc67QeaPbiResoAYIUl9V3mAhp7XR4pQ5we1KcuL8,4553
 official/nlp/modeling/networks/funnel_transformer.py,sha256=5_dcVgOgJoXZ-wZT2X5w-yaNt1OTqj-whWjg73bKkSg,24127
 official/nlp/modeling/networks/funnel_transformer_test.py,sha256=XEVGmOa-sy7CTe46NQkQ18MZGdWTEB-5qYlujOndCl4,17562
-official/nlp/modeling/networks/mobile_bert_encoder.py,sha256=P2RIMlJm2hhqy6giWecR39G9RLuucsko37ZfP9glsLg,8621
+official/nlp/modeling/networks/mobile_bert_encoder.py,sha256=TNzB1ytpeGFB2zrMVZZNPzC0Y-KyelT7QuIpt0EGi7A,8662
 official/nlp/modeling/networks/mobile_bert_encoder_test.py,sha256=u9WqDiIuKlQuA2cEUg_InpgaXXQASGHGhxoCseyHOZI,7115
 official/nlp/modeling/networks/packed_sequence_embedding.py,sha256=Hu1hcTtfBA96dboCNcAPwe2D7x0QYWHxY4BpO5nUDUw,12810
 official/nlp/modeling/networks/packed_sequence_embedding_test.py,sha256=atjMoYpt78Uz16NXdp7l7IrJQtgcfxvkzcKDvhl5bac,5076
@@ -1222,9 +1222,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=nc6A9K53OGqF25xN5St8EiWvdVbda
 tensorflow_models/nlp/__init__.py,sha256=4tA5Pf4qaFwT-fIFOpX7x7FHJpnyJT-5UgOeFYTyMlc,807
 tensorflow_models/uplift/__init__.py,sha256=mqfa55gweOdpKoaQyid4A_4u7xw__FcQeSIF0k_pYmI,999
 tensorflow_models/vision/__init__.py,sha256=zBorY_v5xva1uI-qxhZO3Qh-Dii-Suq6wEYh6hKHDfc,833
-tf_models_nightly-2.18.0.dev20240903.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.18.0.dev20240903.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.18.0.dev20240903.dist-info/METADATA,sha256=Jqhpg4PHlXh6Y3uOIg7ilbbBDcDZJo-z9l25ed2FYiQ,1432
-tf_models_nightly-2.18.0.dev20240903.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.18.0.dev20240903.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.18.0.dev20240903.dist-info/RECORD,,
+tf_models_nightly-2.18.0.dev20240904.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.18.0.dev20240904.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.18.0.dev20240904.dist-info/METADATA,sha256=H8BvDQMEE-qcvmrWfICbmO1Qzv0GVCPM_Mk5BJgvY1M,1432
+tf_models_nightly-2.18.0.dev20240904.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.18.0.dev20240904.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.18.0.dev20240904.dist-info/RECORD,,