Skip to content

Commit b02c73a

Browse files
tastelikefeetJintao-Huang
authored andcommitted
fix qwen3 emb template (#4909)
1 parent 8e77ce9 commit b02c73a

File tree

1 file changed

+2
-1
lines changed
  • swift/llm/template/template

1 file changed

+2
-1
lines changed

swift/llm/template/template/llm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,8 @@
3232
))
3333

3434
register_template(QwenTemplateMeta(MLLMTemplateType.qwen2_gme, template_cls=Qwen2VLTemplate, suffix=['<|endoftext|>']))
35-
register_template(TemplateMeta(LLMTemplateType.qwen3_emb, suffix=[], prefix=[], chat_sep=[], prompt=['{{QUERY}}']))
35+
register_template(
36+
TemplateMeta(LLMTemplateType.qwen3_emb, suffix=['<|endoftext|>'], prefix=[], chat_sep=[], prompt=['{{QUERY}}']))
3637

3738
register_template(
3839
TemplateMeta(LLMTemplateType.baichuan, prefix=['{{SYSTEM}}'], prompt=[[195], '{{QUERY}}', [196]], chat_sep=[]))

0 commit comments

Comments
 (0)