ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix imports

Browse Source 
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
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