"vscode:/vscode.git/clone" did not exist on "ab7857019a438958a0c5c891c22439e1c9de9ec4"
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,6 +226,7 @@ class BertEncoder(tf.keras.Model):
self._embedding_layer = embedding_layer_inst
self._position_embedding_layer = position_embedding_layer
self._type_embedding_layer = type_embedding_layer
if embedding_projection is not None:
self._embedding_projection = embedding_projection
def get_embedding_table(self):
......
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