Update README.md
Browse files
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
|
21 |
|
22 |
base_model = "decapoda-research/llama-7b-hf"
|
23 |
-
tokenizer =
|
24 |
-
model =
|
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",
|