zhifu gao
2023-03-06 679ee2ee2297a1cab3c1bc4c9ea32c31d3bafbf3
funasr/runtime/triton_gpu/model_repo_paraformer_large_offline/scoring/1/model.py
@@ -22,8 +22,6 @@
import json
import os
import pickle
class TritonPythonModel:
    """Your Python model must use the same class name. Every Python model
    that is created must have "TritonPythonModel" as the class name.
@@ -75,8 +73,8 @@
        """
        load lang_char.txt
        """
        with open(str(vocab_file), 'rb') as f:
            token_list = pickle.load(f)
        with open(str(vocab_file), 'r') as f:
            token_list = [line.strip() for line in f]
        return token_list
    def execute(self, requests):