ayousanz commited on
Commit
288cb59
1 Parent(s): b5b4569

Add ONNX and ORT models with quantization

Browse files
.gitattributes CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ ort_models/model.ort filter=lfs diff=lfs merge=lfs -text
37
+ ort_models/model_fp16.ort filter=lfs diff=lfs merge=lfs -text
38
+ ort_models/model_int8.ort filter=lfs diff=lfs merge=lfs -text
39
+ ort_models/model_uint8.ort filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - onnx
5
+ - ort
6
+ ---
7
+
8
+ # ONNX and ORT models with quantization of [google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased)
9
+
10
+ [日本語READMEはこちら](README_ja.md)
11
+
12
+ This repository contains the ONNX and ORT formats of the model [google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased), along with quantized versions.
13
+
14
+ ## License
15
+ The license for this model is "mit". For details, please refer to the original model page: [google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased).
16
+
17
+ ## Usage
18
+ To use this model, install ONNX Runtime and perform inference as shown below.
19
+ ```python
20
+ # Example code
21
+ import onnxruntime as ort
22
+ import numpy as np
23
+ from transformers import AutoTokenizer
24
+ import os
25
+
26
+ # Load the tokenizer
27
+ tokenizer = AutoTokenizer.from_pretrained('google-bert/bert-base-german-cased')
28
+
29
+ # Prepare inputs
30
+ text = 'Replace this text with your input.'
31
+ inputs = tokenizer(text, return_tensors='np')
32
+
33
+ # Specify the model paths
34
+ # Test both the ONNX model and the ORT model
35
+ model_paths = [
36
+ 'onnx_models/model_opt.onnx', # ONNX model
37
+ 'ort_models/model.ort' # ORT format model
38
+ ]
39
+
40
+ # Run inference with each model
41
+ for model_path in model_paths:
42
+ print(f'\n===== Using model: {model_path} =====')
43
+ # Get the model extension
44
+ model_extension = os.path.splitext(model_path)[1]
45
+
46
+ # Load the model
47
+ if model_extension == '.ort':
48
+ # Load the ORT format model
49
+ session = ort.InferenceSession(model_path, providers=['CPUExecutionProvider'])
50
+ else:
51
+ # Load the ONNX model
52
+ session = ort.InferenceSession(model_path)
53
+
54
+ # Run inference
55
+ outputs = session.run(None, dict(inputs))
56
+
57
+ # Display the output shapes
58
+ for idx, output in enumerate(outputs):
59
+ print(f'Output {idx} shape: {output.shape}')
60
+
61
+ # Display the results (add further processing if needed)
62
+ print(outputs)
63
+ ```
64
+
65
+ ## Contents of the Model
66
+ This repository includes the following models:
67
+
68
+ ### ONNX Models
69
+ - `onnx_models/model.onnx`: Original ONNX model converted from [google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased)
70
+ - `onnx_models/model_opt.onnx`: Optimized ONNX model
71
+ - `onnx_models/model_fp16.onnx`: FP16 quantized model
72
+ - `onnx_models/model_int8.onnx`: INT8 quantized model
73
+ - `onnx_models/model_uint8.onnx`: UINT8 quantized model
74
+
75
+ ### ORT Models
76
+ - `ort_models/model.ort`: ORT model using the optimized ONNX model
77
+ - `ort_models/model_fp16.ort`: ORT model using the FP16 quantized model
78
+ - `ort_models/model_int8.ort`: ORT model using the INT8 quantized model
79
+ - `ort_models/model_uint8.ort`: ORT model using the UINT8 quantized model
80
+
81
+ ## Notes
82
+ Please adhere to the license and usage conditions of the original model [google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased).
83
+
84
+ ## Contribution
85
+ If you find any issues or have improvements, please create an issue or submit a pull request.
README_ja.md ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - onnx
5
+ - ort
6
+ ---
7
+
8
+ # [google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased) のONNXおよびORTモデルと量子化モデル
9
+
10
+ [Click here for the English README](README.md)
11
+
12
+ このリポジトリは、元のモデル [google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased) をONNXおよびORT形式に変換し、さらに量子化したものです。
13
+
14
+ ## ライセンス
15
+ このモデルのライセンスは「mit」です。詳細は元のモデルページ([google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased))を参照してください。
16
+
17
+ ## 使い方
18
+ このモデルを使用するには、ONNX Runtimeをインストールし、以下のように推論を行います。
19
+ ```python
20
+ # サンプルコード
21
+ import onnxruntime as ort
22
+ import numpy as np
23
+ from transformers import AutoTokenizer
24
+ import os
25
+
26
+ # トークナイザーの読み込み
27
+ tokenizer = AutoTokenizer.from_pretrained('google-bert/bert-base-german-cased')
28
+
29
+ # 入力の準備
30
+ text = 'ここに入力テキストを置き換えてください。'
31
+ inputs = tokenizer(text, return_tensors='np')
32
+
33
+ # 使用するモデルのパスを指定
34
+ # ONNXモデルとORTモデルの両方をテストする
35
+ model_paths = [
36
+ 'onnx_models/model_opt.onnx', # ONNXモデル
37
+ 'ort_models/model.ort' # ORTフォーマットのモデル
38
+ ]
39
+
40
+ # モデルごとに推論を実行
41
+ for model_path in model_paths:
42
+ print(f'\n===== Using model: {model_path} =====')
43
+ # モデルの拡張子を取得
44
+ model_extension = os.path.splitext(model_path)[1]
45
+
46
+ # モデルの読み込み
47
+ if model_extension == '.ort':
48
+ # ORTフォーマットのモデルをロード
49
+ session = ort.InferenceSession(model_path, providers=['CPUExecutionProvider'])
50
+ else:
51
+ # ONNXモデルをロード
52
+ session = ort.InferenceSession(model_path)
53
+
54
+ # 推論の実行
55
+ outputs = session.run(None, dict(inputs))
56
+
57
+ # 出力の形状を表示
58
+ for idx, output in enumerate(outputs):
59
+ print(f'Output {idx} shape: {output.shape}')
60
+
61
+ # 結果の表示(必要に応じて処理を追加)
62
+ print(outputs)
63
+ ```
64
+
65
+ ## モデルの内容
66
+ このリポジトリには、以下のモデルが含まれています。
67
+
68
+ ### ONNXモデル
69
+ - `onnx_models/model.onnx`: [google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased) から変換された元のONNXモデル
70
+ - `onnx_models/model_opt.onnx`: 最適化されたONNXモデル
71
+ - `onnx_models/model_fp16.onnx`: FP16による量子化モデル
72
+ - `onnx_models/model_int8.onnx`: INT8による量子化モデル
73
+ - `onnx_models/model_uint8.onnx`: UINT8による量子化モデル
74
+
75
+ ### ORTモデル
76
+ - `ort_models/model.ort`: 最適化されたONNXモデルを使用したORTモデル
77
+ - `ort_models/model_fp16.ort`: FP16量子化モデルを使用したORTモデル
78
+ - `ort_models/model_int8.ort`: INT8量子化モデルを使用したORTモデル
79
+ - `ort_models/model_uint8.ort`: UINT8量子化モデルを使用したORTモデル
80
+
81
+ ## 注意事項
82
+ 元のモデル [google-bert/bert-base-german-cased](https://huggingface.co/google-bert/bert-base-german-cased) のライセンスおよび使用条件を遵守してください。
83
+
84
+ ## 貢献
85
+ 問題や改善点があれば、Issueを作成するかプルリクエストを送ってください。
onnx_models/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cccc3cf11983f8e7bc14db3f66cbe10023788dd2583b170e27ffb51c8eb1034e
3
+ size 436542660
onnx_models/model_fp16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:988c061e030229fb6d733998ed443bf1a8195f6c3eaa29328dbdc374d6aa719f
3
+ size 218415871
onnx_models/model_int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fea4fcfcb1b10d01b8ca51fafa0c7fe0af5027f05f6396d846906adcd9141707
3
+ size 109792968
onnx_models/model_opt.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:777d492bec973359f20b90b1fc011ef04d71e583ba5efa250b0ea18cf71c97b2
3
+ size 436516777
onnx_models/model_uint8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aea61b85b49d9481b4a4249b2cd1ec915e1479633c477d5cf8e1b8d0e00dafd3
3
+ size 109793011
ort_models/model.ort ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b35eb19d26c3c68113fff9dc44ba8dd17e791e66bda4e77fdd1c4e3191710b08
3
+ size 436711168
ort_models/model_fp16.ort ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6974c48c3fddd06b4897f011443c160b6ca640b404f2f1b0a69fb96c406bb6f
3
+ size 218964304
ort_models/model_int8.ort ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb062569b8f3e9ea9be7d7a47eac97712795dda9b8198bbd713f7e11edac39cb
3
+ size 109957848
ort_models/model_uint8.ort ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59f398c68296c00cfd051f0a22346db5726739618faed82170722ab09c4c9686
3
+ size 109957848