423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-04 20:52:59 +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 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

2
src/llmtuner/webui/components/export.py
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

2
src/llmtuner/webui/components/top.py
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