Sfoglia il codice sorgente

update paddlex_cli

cuicheng01 10 mesi fa
parent
commit
ad43de32eb
1 ha cambiato i file con 8 aggiunte e 6 eliminazioni
  1. 8 6
      paddlex/paddlex_cli.py

+ 8 - 6
paddlex/paddlex_cli.py

@@ -280,12 +280,14 @@ def main():
             interactive_get_pipeline(args.get_pipeline_config, args.save_path)
         else:
             pipeline_args_dict = {}
-            for arg in pipeline_args:
-                arg_name = arg["name"].lstrip("-")
-                if hasattr(args, arg_name):
-                    pipeline_args_dict[arg_name] = getattr(args, arg_name)
-                else:
-                    logging.warning(f"Argument {arg_name} is missing in args")
+            from .utils.flags import USE_NEW_INFERENCE
+            if USE_NEW_INFERENCE:
+                for arg in pipeline_args:
+                    arg_name = arg["name"].lstrip("-")
+                    if hasattr(args, arg_name):
+                        pipeline_args_dict[arg_name] = getattr(args, arg_name)
+                    else:
+                        logging.warning(f"Argument {arg_name} is missing in args")
             return pipeline_predict(
                 args.pipeline,
                 args.input,