Sfoglia il codice sorgente

fix lm_inference_launch.py

wucong.lyb 2 anni fa
parent
commit
42e1259075
1 ha cambiato i file con 2 aggiunte e 1 eliminazioni
  1. 2 1
      funasr/bin/lm_inference_launch.py

+ 2 - 1
funasr/bin/lm_inference_launch.py

@@ -401,7 +401,8 @@ def main(cmd=None):
 
 
     kwargs.pop("gpuid_list", None)
     kwargs.pop("gpuid_list", None)
     kwargs.pop("njob", None)
     kwargs.pop("njob", None)
-    results = inference_launch(**kwargs)
+    inference_pipeline = inference_launch(**kwargs)
+    return inference_pipeline(kwargs["data_path_and_name_and_type"])
 
 
 
 
 if __name__ == "__main__":
 if __name__ == "__main__":