BelleGroup
/

BELLE-LLaMA-13B-2M-enc

Text2Text Generation

Model card Files Files and versions Community

mabaochang commited on Apr 12, 2023

Commit

f9a0f52

•

1 Parent(s): 39a408a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -85,7 +85,7 @@ After you decrypt the files, BELLE-LLAMA-13B-2M can be easily loaded with LlamaF
 from transformers import LlamaForCausalLM, AutoTokenizer
 import torch
-ckpt = './result/BELLE-LLAMA-13B-2M'
 device = torch.device('cuda')
 model = LlamaForCausalLM.from_pretrained(ckpt, device_map='auto', low_cpu_mem_usage=True)
 tokenizer = AutoTokenizer.from_pretrained(ckpt)

 from transformers import LlamaForCausalLM, AutoTokenizer
 import torch
+ckpt = './result/'
 device = torch.device('cuda')
 model = LlamaForCausalLM.from_pretrained(ckpt, device_map='auto', low_cpu_mem_usage=True)
 tokenizer = AutoTokenizer.from_pretrained(ckpt)