update readme

This commit is contained in:
hiyouga committed 2024-04-22 17:09:17 +08:00
1 parent 836ca05586
commit db7f3b9784
3 files changed
+9 -2

No files matched your search

+3
View File
@@ -94,6 +94,9 @@ def _check_extra_dependencies(
if finetuning_args.use_badam:
require_version("badam", "To fix: pip install badam")
if finetuning_args.plot_loss:
require_version("matplotlib", "To fix: pip install matplotlib")
if training_args is not None and training_args.predict_with_generate:
require_version("jieba", "To fix: pip install jieba")
require_version("nltk", "To fix: pip install nltk")