Update README.md
Browse files
README.md
CHANGED
@@ -36,6 +36,16 @@ language:
|
|
36 |
|
37 |
Use the code below to get started with the model.
|
38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
[More Information Needed]
|
40 |
|
41 |
## Training Details
|
|
|
36 |
|
37 |
Use the code below to get started with the model.
|
38 |
|
39 |
+
```python
|
40 |
+
from peft import PeftModel, PeftConfig
|
41 |
+
from transformers import AutoModelForCausalLM
|
42 |
+
|
43 |
+
config = PeftConfig.from_pretrained("barbaroo/gptsw3_lora_fo_40b")
|
44 |
+
model = AutoModelForCausalLM.from_pretrained("AI-Sweden-Models/gpt-sw3-40b")
|
45 |
+
model = PeftModel.from_pretrained(model, "barbaroo/gptsw3_lora_fo_40b")
|
46 |
+
|
47 |
+
```
|
48 |
+
|
49 |
[More Information Needed]
|
50 |
|
51 |
## Training Details
|