Skip to content

Commit 6e8bf84

Browse files
authored
Update py interface for GLM4 (#306)
1 parent 1003af9 commit 6e8bf84

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

.gitignore

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@ dist/
1414
.hypothesis/
1515
.venv
1616

17+
chatglm_cpp/__init__.pyi
18+
1719
# cpp
1820
build/
1921

chatglm_cpp/_C.pyi

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ ChatGLM.cpp python binding
33
"""
44
from __future__ import annotations
55
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']
77
class Baichuan13BForCausalLM(BaseModelForCausalLM):
88
pass
99
class Baichuan7BForCausalLM(BaseModelForCausalLM):
@@ -31,6 +31,8 @@ class ChatGLM2Tokenizer(BaseTokenizer):
3131
pass
3232
class ChatGLM3Tokenizer(BaseTokenizer):
3333
pass
34+
class ChatGLM4Tokenizer(BaseTokenizer):
35+
pass
3436
class ChatGLMForCausalLM(BaseModelForCausalLM):
3537
pass
3638
class 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

Comments
 (0)