fix imports

Former-commit-id: 6156f1abef631c675d150dd1cb0325cfc3820c91
This commit is contained in:
hiyouga
2023-11-15 16:47:45 +08:00
parent 09a4474e7f
commit 8079584143
3 changed files with 3 additions and 3 deletions

View File

@@ -1,7 +1,7 @@
import gradio as gr
from typing import TYPE_CHECKING, Dict, Generator, List
from llmtuner.tuner import export_model
from llmtuner.train import export_model
from llmtuner.webui.common import get_save_dir
from llmtuner.webui.locales import ALERTS

View File

@@ -1,8 +1,8 @@
import gradio as gr
from typing import TYPE_CHECKING, Dict
from llmtuner.data.template import templates
from llmtuner.extras.constants import METHODS, SUPPORTED_MODELS
from llmtuner.extras.template import templates
from llmtuner.webui.common import get_model_path, get_template, list_checkpoint, save_config
from llmtuner.webui.utils import can_quantize