speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add extract_token binary

Browse Source
This commit is contained in:
志浩 2024-09-24 23:10:49 +08:00
parent 342d781f0a
commit 2892a70cd8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/bin/extract_token.py
View File

@ -42,7 +42,7 @@ def main_hydra(cfg: DictConfig):
torch.cuda.set_device(int(os.environ["LOCAL_RANK"]))
model = AutoModel(**kwargs)
res = model.generate(input=kwargs["input"])
print(res)
logging.info(f"{res}")
if __name__ == "__main__":