add web demo

Former-commit-id: 25ea647e5ac36b497b8e176b123fdee39be3fd30
This commit is contained in:
hiyouga
2023-07-18 17:21:16 +08:00
parent 5ed2a97056
commit 8e4ae0aaac
4 changed files with 59 additions and 4 deletions

View File

@@ -11,10 +11,12 @@ from llmtuner.webui.locales import ALERTS
class WebChatModel(ChatModel):
def __init__(self):
def __init__(self, *args):
self.model = None
self.tokenizer = None
self.generating_args = GeneratingArguments()
if len(args) != 0:
super().__init__(*args)
def load_model(
self,