@@ -3,7 +3,7 @@ ChatGLM.cpp python binding
3
3
"""
4
4
from __future__ import annotations
5
5
import typing
6
- __all__ = ['Baichuan13BForCausalLM' , 'Baichuan7BForCausalLM' , 'BaichuanTokenizer' , 'BaseModelForCausalLM' , 'BaseTokenizer' , 'ChatGLM2ForCausalLM' , 'ChatGLM2Tokenizer' , 'ChatGLM3Tokenizer' , 'ChatGLMForCausalLM' , 'ChatGLMTokenizer' , 'ChatMessage' , 'CodeMessage' , 'FunctionMessage' , 'GenerationConfig' , 'InternLMForCausalLM' , 'InternLMTokenizer' , 'ModelConfig' , 'ModelType' , 'Pipeline' , 'ToolCallMessage' ]
6
+ __all__ = ['Baichuan13BForCausalLM' , 'Baichuan7BForCausalLM' , 'BaichuanTokenizer' , 'BaseModelForCausalLM' , 'BaseTokenizer' , 'ChatGLM2ForCausalLM' , 'ChatGLM2Tokenizer' , 'ChatGLM3Tokenizer' , 'ChatGLM4Tokenizer' , ' ChatGLMForCausalLM' , 'ChatGLMTokenizer' , 'ChatMessage' , 'CodeMessage' , 'FunctionMessage' , 'GenerationConfig' , 'InternLMForCausalLM' , 'InternLMTokenizer' , 'ModelConfig' , 'ModelType' , 'Pipeline' , 'ToolCallMessage' ]
7
7
class Baichuan13BForCausalLM (BaseModelForCausalLM ):
8
8
pass
9
9
class Baichuan7BForCausalLM (BaseModelForCausalLM ):
@@ -31,6 +31,8 @@ class ChatGLM2Tokenizer(BaseTokenizer):
31
31
pass
32
32
class ChatGLM3Tokenizer (BaseTokenizer ):
33
33
pass
34
+ class ChatGLM4Tokenizer (BaseTokenizer ):
35
+ pass
34
36
class ChatGLMForCausalLM (BaseModelForCausalLM ):
35
37
pass
36
38
class ChatGLMTokenizer (BaseTokenizer ):
@@ -134,6 +136,8 @@ class ModelType:
134
136
135
137
CHATGLM3
136
138
139
+ CHATGLM4
140
+
137
141
BAICHUAN7B
138
142
139
143
BAICHUAN13B
@@ -145,8 +149,9 @@ class ModelType:
145
149
CHATGLM : typing .ClassVar [ModelType ] # value = <ModelType.CHATGLM: 1>
146
150
CHATGLM2 : typing .ClassVar [ModelType ] # value = <ModelType.CHATGLM2: 2>
147
151
CHATGLM3 : typing .ClassVar [ModelType ] # value = <ModelType.CHATGLM3: 3>
152
+ CHATGLM4 : typing .ClassVar [ModelType ] # value = <ModelType.CHATGLM4: 4>
148
153
INTERNLM : typing .ClassVar [ModelType ] # value = <ModelType.INTERNLM: 1280>
149
- __members__ : typing .ClassVar [dict [str , ModelType ]] # value = {'CHATGLM': <ModelType.CHATGLM: 1>, 'CHATGLM2': <ModelType.CHATGLM2: 2>, 'CHATGLM3': <ModelType.CHATGLM3: 3>, 'BAICHUAN7B': <ModelType.BAICHUAN7B: 1024>, 'BAICHUAN13B': <ModelType.BAICHUAN13B: 1025>, 'INTERNLM': <ModelType.INTERNLM: 1280>}
154
+ __members__ : typing .ClassVar [dict [str , ModelType ]] # value = {'CHATGLM': <ModelType.CHATGLM: 1>, 'CHATGLM2': <ModelType.CHATGLM2: 2>, 'CHATGLM3': <ModelType.CHATGLM3: 3>, 'CHATGLM4': <ModelType.CHATGLM4: 4>, ' BAICHUAN7B': <ModelType.BAICHUAN7B: 1024>, 'BAICHUAN13B': <ModelType.BAICHUAN13B: 1025>, 'INTERNLM': <ModelType.INTERNLM: 1280>}
150
155
def __eq__ (self , other : typing .Any ) -> bool :
151
156
...
152
157
def __getstate__ (self ) -> int :
0 commit comments