@@ -3,7 +3,7 @@ ChatGLM.cpp python binding
33"""
44from __future__ import annotations
55import 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' ]
77class Baichuan13BForCausalLM (BaseModelForCausalLM ):
88 pass
99class Baichuan7BForCausalLM (BaseModelForCausalLM ):
@@ -31,6 +31,8 @@ class ChatGLM2Tokenizer(BaseTokenizer):
3131 pass
3232class ChatGLM3Tokenizer (BaseTokenizer ):
3333 pass
34+ class ChatGLM4Tokenizer (BaseTokenizer ):
35+ pass
3436class ChatGLMForCausalLM (BaseModelForCausalLM ):
3537 pass
3638class ChatGLMTokenizer (BaseTokenizer ):
@@ -134,6 +136,8 @@ class ModelType:
134136
135137 CHATGLM3
136138
139+ CHATGLM4
140+
137141 BAICHUAN7B
138142
139143 BAICHUAN13B
@@ -145,8 +149,9 @@ class ModelType:
145149 CHATGLM : typing .ClassVar [ModelType ] # value = <ModelType.CHATGLM: 1>
146150 CHATGLM2 : typing .ClassVar [ModelType ] # value = <ModelType.CHATGLM2: 2>
147151 CHATGLM3 : typing .ClassVar [ModelType ] # value = <ModelType.CHATGLM3: 3>
152+ CHATGLM4 : typing .ClassVar [ModelType ] # value = <ModelType.CHATGLM4: 4>
148153 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>}
150155 def __eq__ (self , other : typing .Any ) -> bool :
151156 ...
152157 def __getstate__ (self ) -> int :
0 commit comments