[Bugfix]: Make chat content text allow type content (#9358)
Signed-off-by: Vinay Damodaran <vrdn@hey.com>
This commit is contained in:
committed by
GitHub
parent
b7df53cd42
commit
33bab41060
@@ -89,6 +89,7 @@ class EngineArgs:
|
||||
task: TaskOption = "auto"
|
||||
skip_tokenizer_init: bool = False
|
||||
tokenizer_mode: str = 'auto'
|
||||
chat_template_text_format: str = 'string'
|
||||
trust_remote_code: bool = False
|
||||
download_dir: Optional[str] = None
|
||||
load_format: str = 'auto'
|
||||
@@ -250,6 +251,14 @@ class EngineArgs:
|
||||
'fast tokenizer if available.\n* "slow" will '
|
||||
'always use the slow tokenizer. \n* '
|
||||
'"mistral" will always use the `mistral_common` tokenizer.')
|
||||
parser.add_argument(
|
||||
'--chat-template-text-format',
|
||||
type=str,
|
||||
default=EngineArgs.chat_template_text_format,
|
||||
choices=['string', 'openai'],
|
||||
help='The format to render text content within a chat template. '
|
||||
'"string" will keep the content field as a string whereas '
|
||||
'"openai" will parse content in the current OpenAI format.')
|
||||
parser.add_argument('--trust-remote-code',
|
||||
action='store_true',
|
||||
help='Trust remote code from huggingface.')
|
||||
@@ -858,6 +867,7 @@ class EngineArgs:
|
||||
# We know this is not None because we set it in __post_init__
|
||||
tokenizer=cast(str, self.tokenizer),
|
||||
tokenizer_mode=self.tokenizer_mode,
|
||||
chat_template_text_format=self.chat_template_text_format,
|
||||
trust_remote_code=self.trust_remote_code,
|
||||
dtype=self.dtype,
|
||||
seed=self.seed,
|
||||
|
||||
Reference in New Issue
Block a user