fix: Include all Dense projection layers in ONNX export (output dim 48) (#3)
Browse files- fix: Include all Dense projection layers in ONNX export (output dim 48) (4451857b361128891d8d6677d2a2f3683eadce8f)
- fix: Re-export ONNX with pylate-onnx-export v1.0.7 (includes all Dense layers, output dim=48) (03eca9b4c6091b7cdfe3fb3a9c036ac4e1cf8555)
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:a2fb3ad410f1b68479caa31e9d6821213050ef25804cb9a3c9ba963e72503f46
|
| 3 |
+
size 67970609
|
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:1d4a2876f8c325f43b97d186e22f6b63ebe9820827d73b81fcea180c20e5518b
|
| 3 |
+
size 17228400
|
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 |
}
|