Update README.md
Browse files
README.md
CHANGED
@@ -6,7 +6,7 @@ language:
|
|
6 |
|
7 |
# Model Card
|
8 |
|
9 |
-
NorGPT-3B-continue-
|
10 |
|
11 |
Prompt format:
|
12 |
```
|
@@ -25,7 +25,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
25 |
import torch
|
26 |
|
27 |
source_model_id = "NorGLM/NorGPT-3B-continue"
|
28 |
-
peft_model_id = "NorGLM/NorGPT-3B-continue-
|
29 |
|
30 |
config = PeftConfig.from_pretrained(peft_model_id)
|
31 |
model = AutoModelForCausalLM.from_pretrained(source_model_id, device_map='balanced')
|
|
|
6 |
|
7 |
# Model Card
|
8 |
|
9 |
+
NorGPT-3B-continue-Instruction-peft is trained on top of [NorGPT-3B-continue](https://huggingface.co/NorGLM/NorGPT-3B-continue) model on [NO-Alpaca](https://huggingface.co/datasets/NbAiLab/norwegian-alpaca) dataset.
|
10 |
|
11 |
Prompt format:
|
12 |
```
|
|
|
25 |
import torch
|
26 |
|
27 |
source_model_id = "NorGLM/NorGPT-3B-continue"
|
28 |
+
peft_model_id = "NorGLM/NorGPT-3B-continue-Instruction-peft"
|
29 |
|
30 |
config = PeftConfig.from_pretrained(peft_model_id)
|
31 |
model = AutoModelForCausalLM.from_pretrained(source_model_id, device_map='balanced')
|