Views
No views yet
1from peft import PeftModel
2from transformers import LLaMATokenizer, LLaMAForCausalLM, GenerationConfig
3
4tokenizer = LLaMATokenizer.from_pretrained("decapoda-research/llama-7b-hf")
5model = LLaMAForCausalLM.from_pretrained(
6 "decapoda-research/llama-7b-hf",
7 load_in_8bit=False,
8 torch_dtype=torch.float16,
9 device_map="auto",
10)
11model = PeftModel.from_pretrained(model, "avocardio/alpaca-lora-7b-german-base-52k")