Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/OpenNMT/CTranslate2.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGuillaume Klein <guillaume.klein@systrangroup.com>2018-12-12 17:19:52 +0300
committerGuillaume Klein <guillaume.klein@systrangroup.com>2018-12-12 17:19:52 +0300
commitb2fb23735ba315a76c7ebb8d62d19afb687ba07b (patch)
tree8fb2cc17e6d2e2651852ac6e2cc90cabbecb5f0c
parentac5970513d48eb24e95b563581c7390025b0de86 (diff)
Improve TensorFlow converter compatibilityv0.10.0
-rw-r--r--python/ctranslate2/converters/opennmt_tf.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/python/ctranslate2/converters/opennmt_tf.py b/python/ctranslate2/converters/opennmt_tf.py
index dc05fb97..773229aa 100644
--- a/python/ctranslate2/converters/opennmt_tf.py
+++ b/python/ctranslate2/converters/opennmt_tf.py
@@ -11,7 +11,7 @@ from ctranslate2.specs import catalog
def load_model(model_dir, src_vocab=None, tgt_vocab=None):
"""Loads variables and vocabularies from a TensorFlow checkpoint or SavedModel."""
- if tf.saved_model.maybe_saved_model_directory(model_dir):
+ if tf.saved_model.loader.maybe_saved_model_directory(model_dir):
config = tf.ConfigProto(device_count={'GPU': 0})
with tf.Session(config=config) as sess:
meta_graph = tf.saved_model.loader.load(sess, ["serve"], model_dir)