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 packages

Browse Source 
Former-commit-id: 8e04794b2da067a4123b9d7091a54c5647f44244
This commit is contained in:
hiyouga 2024-03-17 22:32:03 +08:00
parent a5537f3ee8
commit 3d483e0914
3 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/llmtuner/train/sft/metric.py
View File

@ -2,6 +2,7 @@ from dataclasses import dataclass
from typing import TYPE_CHECKING, Dict, Sequence, Tuple, Union
import numpy as np
from transformers.utils.versions import require_version
from ...extras.constants import IGNORE_INDEX
from ...extras.packages import is_jieba_available, is_nltk_available, is_rouge_available
@ -32,6 +33,10 @@ class ComputeMetrics:
r"""
Uses the model predictions to compute metrics.
"""
require_version("jieba", "To fix: pip install jieba")
require_version("nltk", "To fix: pip install nltk")
require_version("rouge_chinese", "To fix: pip install rouge-chinese")
preds, labels = eval_preds
score_dict = {"rouge-1": [], "rouge-2": [], "rouge-l": [], "bleu-4": []}

2
src/llmtuner/train/utils.py
View File

@ -160,7 +160,7 @@ def _create_galore_optimizer(
training_args: "Seq2SeqTrainingArguments",
finetuning_args: "FinetuningArguments",
) -> "torch.optim.Optimizer":
require_version("galore_torch", "To fix: pip install git+https://github.com/hiyouga/GaLore.git")
require_version("galore_torch", "To fix: pip install galore-torch")
if len(finetuning_args.galore_target) == 1 and finetuning_args.galore_target[0] == "all":
galore_targets = find_all_linear_modules(model)