From e1962169503546f4697508ca4cb4412ccdb785f4 Mon Sep 17 00:00:00 2001 From: Frank Liu Date: Wed, 23 Aug 2023 21:52:39 -0700 Subject: [PATCH] [tokenizer] Print out warning in model_zoo_importer (#2759) --- extensions/tokenizers/src/main/python/model_zoo_importer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/extensions/tokenizers/src/main/python/model_zoo_importer.py b/extensions/tokenizers/src/main/python/model_zoo_importer.py index 2841dbde452..9ed32ec58ef 100644 --- a/extensions/tokenizers/src/main/python/model_zoo_importer.py +++ b/extensions/tokenizers/src/main/python/model_zoo_importer.py @@ -41,6 +41,8 @@ def main(): temp_dir = f"{args.output_dir}/tmp" models = huggingface_models.list_models(args) + if not models: + logging.warning(f"model not found: {args}") for model in models: task = model["task"]