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')
|