"examples/pytorch/vscode:/vscode.git/clone" did not exist on "0526b885ee97a70933cc6cf0e5daef8f9e3038ed"
Commit 463dc2d0 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Do not create embedding projection attr if it is None, for backwards compatibility.

PiperOrigin-RevId: 338786173
parent 589f0269
...@@ -226,7 +226,8 @@ class BertEncoder(tf.keras.Model): ...@@ -226,7 +226,8 @@ class BertEncoder(tf.keras.Model):
self._embedding_layer = embedding_layer_inst self._embedding_layer = embedding_layer_inst
self._position_embedding_layer = position_embedding_layer self._position_embedding_layer = position_embedding_layer
self._type_embedding_layer = type_embedding_layer self._type_embedding_layer = type_embedding_layer
self._embedding_projection = embedding_projection if embedding_projection is not None:
self._embedding_projection = embedding_projection
def get_embedding_table(self): def get_embedding_table(self):
return self._embedding_layer.embeddings return self._embedding_layer.embeddings
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment