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

add ziya prompt template

Browse Source 
Former-commit-id: 321e44ac54a91260cf00a4caa1991708814473fc
This commit is contained in:
hiyouga
2023-06-03 19:05:51 +08:00
parent 5eef8d5d98
commit fa850ae6e5
6 changed files with 79 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/export_model.py
View File

@@ -3,19 +3,15 @@
# Usage: python export_model.py --checkpoint_dir path_to_checkpoint --output_dir path_to_save_model
from transformers import HfArgumentParser, TrainingArguments
from utils import ModelArguments, FinetuningArguments, load_pretrained
from utils import load_pretrained, prepare_args
def main():
parser = HfArgumentParser((ModelArguments, TrainingArguments, FinetuningArguments))
model_args, training_args, finetuning_args = parser.parse_args_into_dataclasses()
model_args, _, training_args, finetuning_args = prepare_args(stage="sft")
model, tokenizer = load_pretrained(model_args, finetuning_args)
model.save_pretrained(training_args.output_dir, max_shard_size="10GB")
tokenizer.save_pretrained(training_args.output_dir)
print("model and tokenizer have been saved at:", training_args.output_dir)