feat: fix bug

e9d9cf62 · chenpangpang · 697496bc · e9d9cf62
Commit e9d9cf62 authored Oct 11, 2024 by chenpangpang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tensorflow/bert/infer.py tensorflow/bert/infer.py +2 -0

No files found.
--- a/tensorflow/bert/infer.py
+++ b/tensorflow/bert/infer.py
+import os
+os.environ['TF_USE_LEGACY_KERAS'] = '1'
 import tensorflow as tf
 import tensorflow_hub as hub
 import tensorflow_text as text