atsuki-yamaguchi commited on
Commit
ee22e7a
1 Parent(s): 28f88ce

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 Gemma2 9B adapted for Telugu using 30K target lang
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/gemma-2-9b-te-30K-lapt"
37
  )
 
38
  tokenizer = AutoTokenizer.from_pretrained(
39
- "atsuki-yamaguchi/gemma-2-9b-te-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
+ "google/gemma-2-9b"
38
+ )
39
+ model = PeftModelForCausalLM.from_pretrained(
40
+ model,
41
  "atsuki-yamaguchi/gemma-2-9b-te-30K-lapt"
42
  )
43
+ model = model.merge_and_unload()
44
  tokenizer = AutoTokenizer.from_pretrained(
45
+ "google/gemma-2-9b"
46
  )
47
  ```
48