-
Notifications
You must be signed in to change notification settings - Fork 301
Update BytePairTokenizerCache to have similar dtypes for x and y in self.factors. #871
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA). View this failed invocation of the CLA check for more information. For the most up to date status, view the checks section at the bottom of the pull request. |
Thanks very much for the contribution! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
One minor comment.
# use the integer key to find the value. | ||
self.factors = tf.pow(256, tf.range(0, 8, dtype=tf.int64)) | ||
self.factors = tf.pow( | ||
tf.constant([256], dtype=tf.int64), tf.range(0, 8, dtype=tf.int64) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why do we need this in a list instead of just tf.constant(256, dtype=tf.int64)
? I don't get why we need to change the shape.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the PR!
+1 to Matt's comment, I think we can drop the []
.
Totally agree. Dropped the |
TY! |
…elf.factors. (keras-team#871) * Update BytePairTokenizerCache to have similar dtypes for x and y in self.factors. * Drop list in tf.constant.
…elf.factors. (keras-team#871) * Update BytePairTokenizerCache to have similar dtypes for x and y in self.factors. * Drop list in tf.constant.
Fix for issue #870.
tf.pow
was invoked with different dtypes resulting in errors when executing in graph mode. Setting the dtype to tf.int64 fixes the issue.