Views
No views yet
1from peft import PeftModel
2from transformers import GenerationConfig, LlamaForCausalLM, LlamaTokenizer
3
4
5tokenizer = LlamaTokenizer.from_pretrained("decapoda-research/llama-7b-hf")
6model = LlamaForCausalLM.from_pretrained(
7 "decapoda-research/llama-7b-hf",
8 load_in_8bit=True,
9 torch_dtype=torch.float16,
10 device_map="auto"
11)
12model = PeftModel.from_pretrained(
13 model,
14 "DataAgent/llama-7b-alpaca-zh-20k",
15 torch_dtype=torch.float16
16)