Hardik commited on
Commit
525fd77
·
1 Parent(s): eafed38

LSTM: try keras.saving.load_model first

Browse files
ml_service/app/services/model_loader.py CHANGED
@@ -108,6 +108,7 @@ class ModelManager:
108
  last_lstm_error = ""
109
 
110
  for loader_fn, name in [
 
111
  (lambda p: keras.models.load_model(p), "keras"),
112
  (lambda p: tf.keras.models.load_model(p, compile=False), "tf.keras"),
113
  ]:
 
108
  last_lstm_error = ""
109
 
110
  for loader_fn, name in [
111
+ (lambda p: keras.saving.load_model(p), "keras_saving"),
112
  (lambda p: keras.models.load_model(p), "keras"),
113
  (lambda p: tf.keras.models.load_model(p, compile=False), "tf.keras"),
114
  ]: