BM-K commited on
Commit
2a55a8d
β€’
1 Parent(s): f32512b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -8,8 +8,8 @@ Current language models usually consist of hundreds of millions of parameters wh
8
  ```python
9
  from transformers import AutoTokenizer, AutoModel
10
 
11
- tokenizer = AutoTokenizer.from_pretrained("BM-K/KoMiniLM") # 23M model
12
- model = AutoModel.from_pretrained("BM-K/KoMiniLM")
13
 
14
  inputs = tokenizer("μ•ˆλ…• 세상아!", return_tensors="pt")
15
  outputs = model(**inputs)
 
8
  ```python
9
  from transformers import AutoTokenizer, AutoModel
10
 
11
+ tokenizer = AutoTokenizer.from_pretrained("BM-K/KoMiniLM-68M") # 68M model
12
+ model = AutoModel.from_pretrained("BM-K/KoMiniLM-68M")
13
 
14
  inputs = tokenizer("μ•ˆλ…• 세상아!", return_tensors="pt")
15
  outputs = model(**inputs)