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

release v0.1.4

Browse Source 
Former-commit-id: 81f84aaf2e120e39edb28ef42893939fc9a184e2
This commit is contained in:
hiyouga
2023-08-01 10:08:47 +08:00
parent 772ad4ec6b
commit 661890b8a1
13 changed files with 66 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/llmtuner/webui/manager.py
View File

@@ -1,15 +1,17 @@
import gradio as gr
from typing import Any, Dict, List
from gradio.components import Component
from typing import TYPE_CHECKING, Any, Dict, List
from llmtuner.webui.common import get_model_path, list_dataset, load_config
from llmtuner.webui.locales import LOCALES
from llmtuner.webui.utils import get_time
if TYPE_CHECKING:
from gradio.components import Component
class Manager:
def __init__(self, elem_list: List[Dict[str, Component]]):
def __init__(self, elem_list: List[Dict[str, "Component"]]):
self.elem_list = elem_list
def gen_refresh(self) -> Dict[str, Any]:
@@ -24,7 +26,7 @@ class Manager:
return refresh_dict
def gen_label(self, lang: str) -> Dict[Component, dict]:
def gen_label(self, lang: str) -> Dict["Component", dict]:
update_dict = {}
refresh_dict = self.gen_refresh()