v-suxuerui
commited on
Commit
·
659fe2d
1
Parent(s):
bcda1a6
bin
Browse files- load_ck.py +5 -0
- pytorch_model.bin +3 -0
load_ck.py
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from transformers import AutoTokenizer, LlamaForCausalLM, LlamaConfig, TrainingArguments ,DataCollatorForLanguageModeling
|
| 2 |
+
import torch
|
| 3 |
+
tokenizer = AutoTokenizer.from_pretrained("./")
|
| 4 |
+
model = LlamaForCausalLM.from_pretrained("./")
|
| 5 |
+
torch.save(model, "pytorch_model.bin")
|
pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:79b1ca35c191debc125c68f44db72b838e42491a1a1067020ee1bb0024cd58af
|
| 3 |
+
size 32121266474
|