funasr/bin/inference.py
@@ -21,11 +21,13 @@ logging.basicConfig(level=log_level) if kwargs.get("debug", False): import pdb; pdb.set_trace() import pdb pdb.set_trace() model = AutoModel(**kwargs) res = model.generate(input=kwargs["input"]) print(res) if __name__ == '__main__': if __name__ == "__main__": main_hydra()