PEFT
barbaroo commited on
Commit
1403a95
·
verified ·
1 Parent(s): 2ce7599

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
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