bookbot
/

domba-lora-v0-1

Model card Files Files and versions Metrics Training metrics Community

w11wo commited on Mar 23, 2023

Commit

7830bbd

•

1 Parent(s): 094ab2a

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -12,10 +12,10 @@ Check the Github repo with code: https://github.com/22-hours/cabrita
 ```python
 from peft import PeftModel
-from transformers import LLaMATokenizer, LLaMAForCausalLM, GenerationConfig
-tokenizer = LLaMATokenizer.from_pretrained("decapoda-research/llama-7b-hf")
-model = LLaMAForCausalLM.from_pretrained(
     "decapoda-research/llama-7b-hf",
     load_in_8bit=True,
     device_map="auto",

 ```python
 from peft import PeftModel
+from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
+tokenizer = LlamaTokenizer.from_pretrained("decapoda-research/llama-7b-hf")
+model = LlamaForCausalLM.from_pretrained(
     "decapoda-research/llama-7b-hf",
     load_in_8bit=True,
     device_map="auto",