fix: Include all Dense projection layers in ONNX export (output dim 64) (#2)
Browse files- fix: Include all Dense projection layers in ONNX export (output dim 64) (e7f3cd9ef4b97f600e8ebb8fc9eecd0d84d982e6)
- fix: Re-export ONNX with pylate-onnx-export v1.0.7 (includes all Dense layers, output dim=64) (601907744aadda4240ce6c2e3e75c64ed4d4cb4f)
Co-authored-by: Chao-Chun (Joe) Hsu <joe32140@users.noreply.huggingface.co>
- model.onnx +2 -2
- model_int8.onnx +2 -2
- onnx_config.json +2 -1
model.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:886e3a1638af8222613a8b3baf73520d5ab8c8275fc5ea16e3166982d01df24e
|
| 3 |
+
size 131487995
|
model_int8.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:264ba680e960af9fffb4f78c3af1e4ff92520678b8e136c79434d88fb2549e1b
|
| 3 |
+
size 33208906
|
onnx_config.json
CHANGED
|
@@ -47,5 +47,6 @@
|
|
| 47 |
"mask_token_id": 50284,
|
| 48 |
"pad_token_id": 50284,
|
| 49 |
"query_prefix_id": 50368,
|
| 50 |
-
"document_prefix_id": 50369
|
|
|
|
| 51 |
}
|
|
|
|
| 47 |
"mask_token_id": 50284,
|
| 48 |
"pad_token_id": 50284,
|
| 49 |
"query_prefix_id": 50368,
|
| 50 |
+
"document_prefix_id": 50369,
|
| 51 |
+
"do_lower_case": true
|
| 52 |
}
|