atsuki-yamaguchi commited on
Commit
db3cf39
1 Parent(s): a2cf17b

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -8,12 +8,13 @@ BLOOM-1B Arabic [LAPT + Heuristics]
8
  ## How to use
9
  ```python
10
  from peft import AutoPeftModelForCausalLM
 
11
 
12
  model = AutoPeftModelForCausalLM.from_pretrained(
13
  "atsuki-yamaguchi/bloom-1b1-heuristics-ar"
14
  )
15
  tokenizer = AutoTokenizer.from_pretrained(
16
- "atsuki-yamaguchi/bloom-1b1-heuristics-ar"
17
  )
18
 
19
  # w/ GPU
 
8
  ## How to use
9
  ```python
10
  from peft import AutoPeftModelForCausalLM
11
+ from transformers import AutoTokenizer
12
 
13
  model = AutoPeftModelForCausalLM.from_pretrained(
14
  "atsuki-yamaguchi/bloom-1b1-heuristics-ar"
15
  )
16
  tokenizer = AutoTokenizer.from_pretrained(
17
+ "aubmindlab/aragpt2-base"
18
  )
19
 
20
  # w/ GPU