funasr/runtime/triton_gpu/model_repo_paraformer_large_offline/scoring/1/model.py
@@ -21,8 +21,7 @@ import json import os import pickle import yaml class TritonPythonModel: """Your Python model must use the same class name. Every Python model @@ -76,8 +75,8 @@ load lang_char.txt """ with open(str(vocab_file), 'rb') as f: token_list = pickle.load(f) return token_list config = yaml.load(f, Loader=yaml.Loader) return config['token_list'] def execute(self, requests): """`execute` must be implemented in every Python model. `execute`
