tf-models-nightly 2.18.0.dev20240908__py2.py3-none-any.whl → 2.18.0.dev20240910__py2.py3-none-any.whl

official/nlp/modeling/networks/mobile_bert_encoder.py

@@ -164,6 +164,36 @@ class MobileBERTEncoder(tf_keras.Model):
         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,
+        max_sequence_length=max_sequence_length,
+        num_blocks=num_blocks,
+        hidden_size=hidden_size,
+        num_attention_heads=num_attention_heads,
+        intermediate_size=intermediate_size,
+        intermediate_act_fn=intermediate_act_fn,
+        hidden_dropout_prob=hidden_dropout_prob,
+        attention_probs_dropout_prob=attention_probs_dropout_prob,
+        intra_bottleneck_size=intra_bottleneck_size,
+        initializer_range=initializer_range,
+        use_bottleneck_attention=use_bottleneck_attention,
+        key_query_shared_bottleneck=key_query_shared_bottleneck,
+        num_feedforward_networks=num_feedforward_networks,
+        normalization_type=normalization_type,
+        classifier_activation=classifier_activation,
+        input_mask_dtype=input_mask_dtype,
+        **kwargs,
+    )
+
+  def get_config(self):
+    return dict(self._config)
+
+  @classmethod
+  def from_config(cls, config):
+    return cls(**config)
 
   def get_embedding_table(self):
     return self.embedding_layer.word_embedding.embeddings

{tf_models_nightly-2.18.0.dev20240908.dist-info → tf_models_nightly-2.18.0.dev20240910.dist-info}/METADATA

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

{tf_models_nightly-2.18.0.dev20240908.dist-info → tf_models_nightly-2.18.0.dev20240910.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=IuAw-nsQFvh4uVgroemaVM7DoclMfeL_BvfKek1qjc4,7542
+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.dev20240908.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.18.0.dev20240908.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.18.0.dev20240908.dist-info/METADATA,sha256=VdAI3hSrgYoAGj0JK4dlRMt3XrbiN65ZoKOEFHF9N0M,1432
-tf_models_nightly-2.18.0.dev20240908.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.18.0.dev20240908.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.18.0.dev20240908.dist-info/RECORD,,
+tf_models_nightly-2.18.0.dev20240910.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.18.0.dev20240910.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.18.0.dev20240910.dist-info/METADATA,sha256=uh_TG9vNe7xJTarAp12mbfYr2n8Z6nTPcxp4sTzYT-c,1432
+tf_models_nightly-2.18.0.dev20240910.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.18.0.dev20240910.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.18.0.dev20240910.dist-info/RECORD,,