add web demo

This commit is contained in:
hiyouga
2023-07-18 17:21:16 +08:00
parent bdf91846da
commit b447fa85aa
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,