[Fix] Fix clean_up_tokenization_spaces in tokenizer (#1510)

This commit is contained in:
Lianmin Zheng
2024-09-24 21:37:33 -07:00
committed by GitHub
parent 067d8e16fc
commit fb2d0680e0
4 changed files with 12 additions and 9 deletions

View File

@@ -129,6 +129,7 @@ def get_tokenizer(
*args,
trust_remote_code=trust_remote_code,
tokenizer_revision=tokenizer_revision,
clean_up_tokenization_spaces=False,
**kwargs,
)
except TypeError as e:

View File

@@ -21,8 +21,9 @@ from typing import List, Union
import torch
import torch.nn.functional as F
from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers import AutoModelForCausalLM
from sglang.srt.hf_transformers_utils import get_tokenizer
from sglang.srt.server import Runtime
from sglang.test.test_utils import DEFAULT_PORT_FOR_SRT_TEST_RUNNER
@@ -92,11 +93,7 @@ class HFRunner:
self.model_proc.start()
def start_model_process(self, in_queue, out_queue, model_path, torch_dtype):
self.tokenizer = AutoTokenizer.from_pretrained(
model_path,
torch_dtype=torch_dtype,
)
self.tokenizer = get_tokenizer(model_path)
if self.is_generation:
self.base_model = AutoModelForCausalLM.from_pretrained(
model_path,