mirror of
https://github.com/JasonYANG170/CodeGeeX4.git
synced 2024-11-27 14:16:35 +00:00
23 lines
375 B
Python
23 lines
375 B
Python
|
"""
|
||
|
|
coding : utf-8
|
||
|
|
@Date : 2024/7/10
|
||
|
|
@Author : Shaobo
|
||
|
|
@Describe:
|
||
|
|
"""
|
||
|
|
from models.codegeex import CodegeexChatModel
|
||
|
|
|
||
|
|
model: CodegeexChatModel
|
||
|
|
|
||
|
|
|
||
|
|
def stream_chat_with_codegeex(request):
|
||
|
|
yield from model.stream_chat(request)
|
||
|
|
|
||
|
|
|
||
|
|
def chat_with_codegeex(request):
|
||
|
|
return model.chat(request)
|
||
|
|
|
||
|
|
|
||
|
|
def init_model(args):
|
||
|
|
global model
|
||
|
|
model = CodegeexChatModel(args)
|