Skip to content

Commit 49c0c91

Browse files
committed
[template] Simplify GLM4_5TemplateMeta class construction
1 parent 46038d6 commit 49c0c91

File tree

1 file changed

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

1 file changed

+1
-8
lines changed

swift/llm/template/template/glm.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -66,15 +66,8 @@ class GLM4_0414TemplateMeta(GLM4TemplateMeta):
6666

6767

6868
@dataclass
69-
class GLM4_5TemplateMeta(GLMTemplateMeta):
70-
prefix: Prompt = field(default_factory=lambda: ['[gMASK]<sop>'])
71-
prompt: Prompt = field(default_factory=lambda: ['<|user|>\n{{QUERY}}<|assistant|>\n'])
72-
chat_sep: Optional[Prompt] = field(default_factory=list)
73-
suffix: Prompt = field(default_factory=lambda: ['<|user|>'])
74-
system_prefix: Optional[Prompt] = field(default_factory=lambda: ['[gMASK]<sop><|system|>\n{{SYSTEM}}'])
75-
69+
class GLM4_5TemplateMeta(GLM4_0414TemplateMeta):
7670
agent_template: str = 'glm4_5'
77-
stop_words: List[Word] = field(default_factory=lambda: ['<|endoftext|>', '<|user|>', '<|observation|>'])
7871

7972
class GLM4_1VTemplateMeta(GLM4_0414TemplateMeta):
8073
system_prefix: Optional[Prompt] = field(default_factory=lambda: ['[gMASK]<sop><|system|>{{SYSTEM}}'])

0 commit comments

Comments
 (0)