fix imports

Former-commit-id: bbbce1f516840f722247edd37057d16502ea0557
This commit is contained in:
hiyouga 2023-11-15 16:47:45 +08:00
parent 06a4820836
commit 4a0be64ae6
3 changed files with 3 additions and 3 deletions

View File

@ -4,8 +4,8 @@ from typing import Any, Dict, Generator, List, Literal, Optional, Tuple
from threading import Thread
from transformers import GenerationConfig, TextIteratorStreamer
from llmtuner.data.template import get_template_and_fix_tokenizer
from llmtuner.extras.misc import get_logits_processor
from llmtuner.extras.template import get_template_and_fix_tokenizer
from llmtuner.model import dispatch_model, get_infer_args, load_model_and_tokenizer

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