pcuenq HF staff commited on
Commit
28b13b0
1 Parent(s): 03883ed

Change both tokenizers

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -33,7 +33,7 @@ def get_prompt(message: str, chat_history: list[tuple[str, str]],
33
 
34
  def get_input_token_length(message: str, chat_history: list[tuple[str, str]], system_prompt: str) -> int:
35
  prompt = get_prompt(message, chat_history, system_prompt)
36
- input_ids = tokenizer([prompt], return_tensors='np', add_special_tokens = False)['input_ids']
37
  return input_ids.shape[-1]
38
 
39
 
@@ -45,7 +45,7 @@ def run(message: str,
45
  top_p: float = 0.95,
46
  top_k: int = 50) -> Iterator[str]:
47
  prompt = get_prompt(message, chat_history, system_prompt)
48
- inputs = tokenizer([prompt], return_tensors='pt').to('cuda')
49
 
50
  streamer = TextIteratorStreamer(tokenizer,
51
  timeout=10.,
 
33
 
34
  def get_input_token_length(message: str, chat_history: list[tuple[str, str]], system_prompt: str) -> int:
35
  prompt = get_prompt(message, chat_history, system_prompt)
36
+ input_ids = tokenizer([prompt], return_tensors='np', add_special_tokens=False)['input_ids']
37
  return input_ids.shape[-1]
38
 
39
 
 
45
  top_p: float = 0.95,
46
  top_k: int = 50) -> Iterator[str]:
47
  prompt = get_prompt(message, chat_history, system_prompt)
48
+ inputs = tokenizer([prompt], return_tensors='pt', add_special_tokens=False).to('cuda')
49
 
50
  streamer = TextIteratorStreamer(tokenizer,
51
  timeout=10.,