423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-20 13:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #3317

Browse Source
This commit is contained in:
hiyouga
2024-04-17 22:17:19 +08:00
parent e3ca57602c
commit 6d641af703
15 changed files with 75 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/llmtuner/webui/components/chatbot.py
View File

@@ -1,11 +1,14 @@
from typing import TYPE_CHECKING, Dict, Tuple
import gradio as gr
from ...data import Role
from ...extras.packages import is_gradio_available
from ..utils import check_json_schema
if is_gradio_available():
import gradio as gr
if TYPE_CHECKING:
from gradio.components import Component

7
src/llmtuner/webui/components/data.py
View File

@@ -2,9 +2,12 @@ import json
import os
from typing import TYPE_CHECKING, Any, Dict, List, Tuple
import gradio as gr
from ...extras.constants import DATA_CONFIG
from ...extras.packages import is_gradio_available
if is_gradio_available():
import gradio as gr
if TYPE_CHECKING:

7
src/llmtuner/webui/components/eval.py
View File

@@ -1,11 +1,14 @@
from typing import TYPE_CHECKING, Dict
import gradio as gr
from ...extras.packages import is_gradio_available
from ..common import DEFAULT_DATA_DIR, list_dataset
from .data import create_preview_box
if is_gradio_available():
import gradio as gr
if TYPE_CHECKING:
from gradio.components import Component

7
src/llmtuner/webui/components/export.py
View File

@@ -1,12 +1,15 @@
from typing import TYPE_CHECKING, Dict, Generator, List
import gradio as gr
from ...extras.packages import is_gradio_available
from ...train import export_model
from ..common import get_save_dir
from ..locales import ALERTS
if is_gradio_available():
import gradio as gr
if TYPE_CHECKING:
from gradio.components import Component

7
src/llmtuner/webui/components/infer.py
View File

@@ -1,10 +1,13 @@
from typing import TYPE_CHECKING, Dict
import gradio as gr
from ...extras.packages import is_gradio_available
from .chatbot import create_chat_box
if is_gradio_available():
import gradio as gr
if TYPE_CHECKING:
from gradio.components import Component

7
src/llmtuner/webui/components/top.py
View File

@@ -1,13 +1,16 @@
from typing import TYPE_CHECKING, Dict
import gradio as gr
from ...data import templates
from ...extras.constants import METHODS, SUPPORTED_MODELS
from ...extras.packages import is_gradio_available
from ..common import get_model_path, get_template, list_adapters, save_config
from ..utils import can_quantize
if is_gradio_available():
import gradio as gr
if TYPE_CHECKING:
from gradio.components import Component

6
src/llmtuner/webui/components/train.py
View File

@@ -1,13 +1,17 @@
from typing import TYPE_CHECKING, Dict
import gradio as gr
from transformers.trainer_utils import SchedulerType
from ...extras.constants import TRAINING_STAGES
from ...extras.packages import is_gradio_available
from ..common import DEFAULT_DATA_DIR, autoset_packing, list_adapters, list_dataset
from ..components.data import create_preview_box
if is_gradio_available():
import gradio as gr
if TYPE_CHECKING:
from gradio.components import Component