| from transformers import AutoTokenizer, LlamaForCausalLM, LlamaConfig, TrainingArguments ,DataCollatorForLanguageModeling | |
| import torch | |
| tokenizer = AutoTokenizer.from_pretrained("./") | |
| model = LlamaForCausalLM.from_pretrained("./") | |
| torch.save(model, "pytorch_model.bin") | |