Skip to content

Conversation

Sruinard
Copy link
Contributor

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.

@google-cla
Copy link

google-cla bot commented Mar 18, 2023

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.

@mattdangerw mattdangerw self-requested a review March 21, 2023 01:04
@mattdangerw
Copy link
Member

Thanks very much for the contribution!

Copy link
Member

@mattdangerw mattdangerw left a 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)
Copy link
Member

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.

Copy link
Contributor

@chenmoneygithub chenmoneygithub left a 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 [].

@Sruinard
Copy link
Contributor Author

Totally agree. Dropped the []. Thanks for the feedback.

@mattdangerw
Copy link
Member

TY!

@mattdangerw mattdangerw merged commit 1e8da37 into keras-team:master Mar 22, 2023
kanpuriyanawab pushed a commit to kanpuriyanawab/keras-nlp that referenced this pull request Mar 26, 2023
…elf.factors. (keras-team#871)

* Update BytePairTokenizerCache to have similar dtypes for x and y in self.factors.

* Drop list in tf.constant.
kanpuriyanawab pushed a commit to kanpuriyanawab/keras-nlp that referenced this pull request Mar 26, 2023
…elf.factors. (keras-team#871)

* Update BytePairTokenizerCache to have similar dtypes for x and y in self.factors.

* Drop list in tf.constant.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants