Commit
•
2517d61
1
Parent(s):
1e3bea9
Update README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,6 @@ from transformers import AutoModelForCausalLM, LlamaTokenizer
|
|
| 31 |
import torch
|
| 32 |
|
| 33 |
ckpt = '/path/to_finetuned_model/'
|
| 34 |
-
ckpt = '/nfs/a100-80G-15/xytian/myProjects/AI_NLP_GM/transformed_models/BELLE2-Llama2-13B-0.4M'
|
| 35 |
device = torch.device('cuda')
|
| 36 |
model = AutoModelForCausalLM.from_pretrained(ckpt).half().to(device)
|
| 37 |
tokenizer = LlamaTokenizer.from_pretrained(ckpt)
|
|
|
|
| 31 |
import torch
|
| 32 |
|
| 33 |
ckpt = '/path/to_finetuned_model/'
|
|
|
|
| 34 |
device = torch.device('cuda')
|
| 35 |
model = AutoModelForCausalLM.from_pretrained(ckpt).half().to(device)
|
| 36 |
tokenizer = LlamaTokenizer.from_pretrained(ckpt)
|