versae commited on
Commit
25279b9
1 Parent(s): e48f137

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -17,11 +17,11 @@ This is an Norwegian adapter generated by fine-tuning LLaMA-7B on a [Norwegian A
17
 
18
  ```python
19
  from peft import PeftModel
20
- from transformers import LLaMATokenizer, LLaMAForCausalLM
21
 
22
  base_model = "decapoda-research/llama-7b-hf"
23
- tokenizer = LLaMATokenizer.from_pretrained(base_model)
24
- model = LLaMAForCausalLM.from_pretrained(
25
  base_model,
26
  load_in_8bit=True,
27
  device_map="auto",
 
17
 
18
  ```python
19
  from peft import PeftModel
20
+ from transformers import LlamaTokenizer, LlamaForCausalLM
21
 
22
  base_model = "decapoda-research/llama-7b-hf"
23
+ tokenizer = LlamaTokenizer.from_pretrained(base_model)
24
+ model = LlamaForCausalLM.from_pretrained(
25
  base_model,
26
  load_in_8bit=True,
27
  device_map="auto",