[djl-convert] Fix huggingface converter (#3505)

deepjavalibrary · Oct 24, 2024 · 0bb604f · 0bb604f
1 parent 77f4226
commit 0bb604f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extensions/tokenizers/src/main/python/djl_converter/huggingface_converter.py b/extensions/tokenizers/src/main/python/djl_converter/huggingface_converter.py
@@ -84,7 +84,7 @@ def save_onnx_model(self, model_info, task: str, args: Namespace,
             sys.argv.extend(["--dtype", args.dtype])
         if args.trust_remote_code:
             sys.argv.append("--trust-remote-code")
-        if os.path.exists(model_id):
+        if task:
             sys.argv.extend(["--task", task])
         sys.argv.append(temp_dir)