Update tf_keras_model.py

2b27ace2 · zhangwenbo · 82619d20 · 2b27ace2
Commit 2b27ace2 authored Feb 04, 2026 by zhangwenbo 🎨
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tf-keras/tf_keras_model.py tf-keras/tf_keras_model.py +2 -0

No files found.
--- a/tf-keras/tf_keras_model.py
+++ b/tf-keras/tf_keras_model.py
+import os
+os.environ['TF_USE_LEGACY_KERAS'] = '1'
 import tensorflow as tf
 from tensorflow import keras