tiktoken做lazyload处理

这个提交包含在:
Your Name
2023-04-19 14:27:34 +08:00
父节点 28aa6d1dc0
当前提交 b0409b929b
共有 12 个文件被更改,包括 83 次插入35 次删除

查看文件

@@ -2,9 +2,9 @@ import traceback
from toolbox import update_ui, get_conf
def input_clipping(inputs, history, max_token_limit):
import tiktoken
import numpy as np
enc = tiktoken.encoding_for_model("gpt-3.5-turbo")
from request_llm.bridge_all import model_info
enc = model_info["gpt-3.5-turbo"]['tokenizer']
def get_token_num(txt): return len(enc.encode(txt, disallowed_special=()))
mode = 'input-and-history'