atsuki-yamaguchi commited on
Commit
fe28432
1 Parent(s): 368b736

Upload README.md with huggingface_hub

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