Views
No views yet
1from peft import PeftModel
2from transformers import LLaMATokenizer, LLaMAForCausalLM, GenerationConfig
3
4tokenizer = LLaMATokenizer.from_pretrained("22h/open-cabrita3b")
5model = LLaMAForCausalLM.from_pretrained(
6 "22h/open-cabrita3b",
7 load_in_8bit=True,
8 device_map="auto",
9)
10model = PeftModel.from_pretrained(model, "jcfneto/lora-cabrita-tv-ptbr")