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

support distributed quantized training

Browse Source 
Former-commit-id: 74ff23a4f36f859f791f7b4be6f1877edc68f12f
This commit is contained in:
hiyouga
2023-06-06 17:39:41 +08:00
parent ac6f50dedf
commit bf5ad34196
7 changed files with 20 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/utils/seq2seq.py
View File

@@ -13,10 +13,10 @@ from nltk.translate.bleu_score import sentence_bleu, SmoothingFunction
from .peft_trainer import PeftTrainer
from .other import get_logger, IGNORE_INDEX
from .other import get_main_logger, IGNORE_INDEX
logger = get_logger(__name__)
logger = get_main_logger(__name__)
@dataclass