Fedor Krasnov
commited on
Commit
·
3649e3b
1
Parent(s):
400b111
alias for quantized
Browse files- README.md +13 -0
- model_quantized.onnx +2 -2
README.md
CHANGED
@@ -74,3 +74,16 @@ unmasker("одежда женское [MASK] для_праздника")
|
|
74 |
'sequence': 'одежда женское белье для_праздника'}]
|
75 |
|
76 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
74 |
'sequence': 'одежда женское белье для_праздника'}]
|
75 |
|
76 |
```
|
77 |
+
|
78 |
+
For transformers.js, it turned out that the ONNX version of the model was required.
|
79 |
+
|
80 |
+
|
81 |
+
```python
|
82 |
+
|
83 |
+
from transformers import AutoTokenizer
|
84 |
+
from optimum.onnxruntime import ORTModelForMaskedLM
|
85 |
+
|
86 |
+
tokenizer = AutoTokenizer.from_pretrained("fkrasnov2/COLD2")
|
87 |
+
model = ORTModelForMaskedLM.from_pretrained("fkrasnov2/COLD2", file_name='model.onnx')
|
88 |
+
|
89 |
+
```
|
model_quantized.onnx
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d82288210bbf54b386814744c44c373568f0f57a218791d9a6440e679b610b42
|
3 |
+
size 39946271
|