ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix webui

Browse Source 
Former-commit-id: 9192948fa221c0275ddfa579ef6b3442d45b8962
This commit is contained in:
hiyouga
2023-11-02 18:03:14 +08:00
parent 12fa56ae68
commit ac74639b32
4 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/llmtuner/webui/chatter.py
View File

@@ -26,9 +26,8 @@ class WebChatModel(ChatModel):
return self.model is not None
def load_model(self, data: Dict[Component, Any]) -> Generator[str, None, None]:
get = lambda name: data[self.manager.get_elem(name)]
get = lambda name: data[self.manager.get_elem_by_name(name)]
lang = get("top.lang")
if self.loaded:
yield ALERTS["err_exists"][lang]
return
@@ -65,9 +64,7 @@ class WebChatModel(ChatModel):
yield ALERTS["info_loaded"][lang]
def unload_model(self, data: Dict[Component, Any]) -> Generator[str, None, None]:
get = lambda name: data[self.manager.get_elem(name)]
lang = get("top.lang")
lang = data[self.manager.get_elem_by_name("top.lang")]
yield ALERTS["info_unloading"][lang]
self.model = None
self.tokenizer = None