qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-05-26 07:39:22 +02:00
Code Issues Projects Releases Wiki Activity

ModelZoo.load_model: fix loading exported vs saved models

Browse source
This commit is contained in:
Robert Sachunsky 2026-05-21 02:39:59 +02:00
parent 3de1407d18
commit 7f2bf715df
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/eynollah/model_zoo/model_zoo.py
View file

@ -191,9 +191,8 @@ class EynollahModelZoo:
try: try:
# avoid wasting VRAM on non-transformer models # avoid wasting VRAM on non-transformer models
model = load_model(model_path, compile=False) model = load_model(model_path, compile=False)
assert isinstance(model, KerasModel)
model.make_predict_function() model.make_predict_function()
except ValueError: except (AttributeError, ValueError):
model = tf.saved_model.load(model_path) model = tf.saved_model.load(model_path)
model.predict_on_batch = model.serve model.predict_on_batch = model.serve
model.input_shape = model.signatures.get('serving_default').inputs[0].shape model.input_shape = model.signatures.get('serving_default').inputs[0].shape