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

fix imports

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/chat/chat_model.py
View File

@ -4,8 +4,8 @@ from typing import Any, Dict, Generator, List, Literal, Optional, Tuple
from threading import Thread from threading import Thread
from transformers import GenerationConfig, TextIteratorStreamer 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.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 from llmtuner.model import dispatch_model, get_infer_args, load_model_and_tokenizer

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

@ -1,7 +1,7 @@
import gradio as gr import gradio as gr
from typing import TYPE_CHECKING, Dict, Generator, List 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.common import get_save_dir
from llmtuner.webui.locales import ALERTS from llmtuner.webui.locales import ALERTS

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

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