[GLM4.6v] Required changes for bumping up to transformer 5.x (#13229)

This commit is contained in:
Binyao Jiang
2025-11-17 18:58:00 -08:00
committed by GitHub
parent d7984f3125
commit 90c18a16cb
11 changed files with 29 additions and 26 deletions

View File

@@ -995,7 +995,10 @@ async def get_mooncake_request_over_time(
# Form the full prompt from history
try:
full_prompt_text = tokenizer.apply_chat_template(
chat_history, tokenize=False, add_generation_prompt=True
chat_history,
tokenize=False,
add_generation_prompt=True,
return_dict=False,
)
except Exception:
full_prompt_text = "\n".join(
@@ -1161,6 +1164,7 @@ def sample_sharegpt_requests(
[{"role": "user", "content": prompt}],
add_generation_prompt=True,
tokenize=False,
return_dict=False,
)
if tokenizer.bos_token:
prompt = prompt.replace(tokenizer.bos_token, "")

View File

@@ -46,7 +46,8 @@ enable_hf_transfer()
class DisabledTqdm(tqdm):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs, disable=True)
kwargs["disable"] = True
super().__init__(*args, **kwargs)
def get_lock(model_name_or_path: str | Path, cache_dir: str | None = None):

View File

@@ -324,6 +324,7 @@ class OpenAIServingChat(OpenAIServingBase):
**(
request.chat_template_kwargs if request.chat_template_kwargs else {}
),
return_dict=False,
)
except Exception:
# This except branch will be triggered when the chosen model
@@ -343,6 +344,7 @@ class OpenAIServingChat(OpenAIServingBase):
**(
request.chat_template_kwargs if request.chat_template_kwargs else {}
),
return_dict=False,
)
if assistant_prefix:

View File

@@ -70,7 +70,8 @@ enable_hf_transfer()
class DisabledTqdm(tqdm):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs, disable=True)
kwargs["disable"] = True
super().__init__(*args, **kwargs)
def get_lock(model_name_or_path: str, cache_dir: Optional[str] = None):

View File

@@ -353,7 +353,7 @@ class HFRunner:
scores = []
for conv in prompts:
conv_formatted = self.tokenizer.apply_chat_template(
conv, tokenize=False
conv, tokenize=False, return_dict=False
)
conv_tokenized = self.tokenizer(
conv_formatted, return_tensors="pt"