support unsloth generate

Former-commit-id: 0ef1ad9f505dba71db9342f524cc3a7565e5e09e
This commit is contained in:
hiyouga
2024-04-24 04:46:53 +08:00
parent 03f2e3284a
commit 5420905a2e
6 changed files with 155 additions and 53 deletions

View File

@@ -61,6 +61,9 @@ def create_modelcard_and_push(
if data_args.dataset is not None:
kwargs["dataset"] = [dataset.strip() for dataset in data_args.dataset.split(",")]
if model_args.use_unsloth:
kwargs["tags"] = kwargs["tags"] + ["unsloth"]
if not training_args.do_train:
pass
elif training_args.push_to_hub: