atsuki-yamaguchi commited on
Commit
e93c4ad
1 Parent(s): 2157bda

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +6 -0
README.md CHANGED
@@ -32,10 +32,16 @@ This model is built on top of Llama2 7B adapted for Sinhala using 30K target lan
32
  Use the code below to get started with the model.
33
  ```python
34
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
35
 
36
  model = AutoModelForCausalLM.from_pretrained(
37
  "atsuki-yamaguchi/Llama-2-7b-hf-si-30K-align-2stage"
38
  )
 
 
 
 
 
39
  tokenizer = AutoTokenizer.from_pretrained(
40
  "atsuki-yamaguchi/Llama-2-7b-hf-si-30K-align-2stage"
41
  )
 
32
  Use the code below to get started with the model.
33
  ```python
34
  from transformers import AutoTokenizer, AutoModelForCausalLM
35
+ from peft import PeftModelForCausalLM
36
 
37
  model = AutoModelForCausalLM.from_pretrained(
38
  "atsuki-yamaguchi/Llama-2-7b-hf-si-30K-align-2stage"
39
  )
40
+ model = PeftModelForCausalLM.from_pretrained(
41
+ model,
42
+ "atsuki-yamaguchi/Llama-2-7b-hf-si-30K-align-2stage"
43
+ )
44
+ model = model.merge_and_unload()
45
  tokenizer = AutoTokenizer.from_pretrained(
46
  "atsuki-yamaguchi/Llama-2-7b-hf-si-30K-align-2stage"
47
  )