File tree Expand file tree Collapse file tree 1 file changed +1
-8
lines changed
swift/llm/template/template Expand file tree Collapse file tree 1 file changed +1
-8
lines changed Original file line number Diff line number Diff line change @@ -66,15 +66,8 @@ class GLM4_0414TemplateMeta(GLM4TemplateMeta):
66
66
67
67
68
68
@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 ):
76
70
agent_template : str = 'glm4_5'
77
- stop_words : List [Word ] = field (default_factory = lambda : ['<|endoftext|>' , '<|user|>' , '<|observation|>' ])
78
71
79
72
class GLM4_1VTemplateMeta (GLM4_0414TemplateMeta ):
80
73
system_prefix : Optional [Prompt ] = field (default_factory = lambda : ['[gMASK]<sop><|system|>{{SYSTEM}}' ])
You can’t perform that action at this time.
0 commit comments