瀏覽代碼

upstream: add

AlpinDale 1 年之前
父節點
當前提交
20a8235114
共有 1 個文件被更改,包括 5 次插入0 次删除
  1. 5 0
      aphrodite/endpoints/llm.py

+ 5 - 0
aphrodite/endpoints/llm.py

@@ -59,6 +59,11 @@ class LLM:
         self,
     ) -> Union[PreTrainedTokenizer, PreTrainedTokenizerFast]:
         return self.aphrodite.tokenizer
+    
+    def set_tokenizer(
+        self,
+        tokenizer: Union[PreTrainedTokenizer, PreTrainedTokenizerFast],
+    ) -> None: self.aphrodite.tokenizer = tokenizer
 
     def generate(
         self,