From fae6fd6d16e0fb060aa063790893f7555c421c4b Mon Sep 17 00:00:00 2001
From: 语帆 <yf352572@alibaba-inc.com>
Date: 星期五, 01 三月 2024 10:29:15 +0800
Subject: [PATCH] atsr
---
funasr/bin/inference.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/funasr/bin/inference.py b/funasr/bin/inference.py
index d2f0c14..3943d7e 100644
--- a/funasr/bin/inference.py
+++ b/funasr/bin/inference.py
@@ -1,7 +1,7 @@
import hydra
import logging
from omegaconf import DictConfig, OmegaConf, ListConfig
-
+import pdb
from funasr.auto.auto_model import AutoModel
@@ -23,6 +23,7 @@
if kwargs.get("debug", False):
import pdb; pdb.set_trace()
model = AutoModel(**kwargs)
+ pdb.set_trace()
res = model.generate(input=kwargs["input"])
print(res)
--
Gitblit v1.9.1