lxyuan mrm8848 commited on
Commit
beaa0ee
·
verified ·
1 Parent(s): 2e33845

Adding ONNX file of this model (#9)

Browse files

- Adding ONNX file of this model (18bef767f699c8f20ed691eaacb5cbdf91c88edc)


Co-authored-by: MrSomething <[email protected]>

Files changed (2) hide show
  1. onnx/config.json +37 -0
  2. onnx/model.onnx +3 -0
onnx/config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_attn_implementation_autoset": true,
3
+ "_name_or_path": "onnx/lxyuan/distilbert-base-multilingual-cased-sentiments-student/checkpoint-100",
4
+ "activation": "gelu",
5
+ "architectures": [
6
+ "DistilBertForSequenceClassification"
7
+ ],
8
+ "attention_dropout": 0.1,
9
+ "dim": 768,
10
+ "dropout": 0.1,
11
+ "hidden_dim": 3072,
12
+ "id2label": {
13
+ "0": "positive",
14
+ "1": "neutral",
15
+ "2": "negative"
16
+ },
17
+ "initializer_range": 0.02,
18
+ "label2id": {
19
+ "negative": 2,
20
+ "neutral": 1,
21
+ "positive": 0
22
+ },
23
+ "max_position_embeddings": 512,
24
+ "model_type": "distilbert",
25
+ "n_heads": 12,
26
+ "n_layers": 6,
27
+ "output_past": true,
28
+ "pad_token_id": 0,
29
+ "problem_type": "single_label_classification",
30
+ "qa_dropout": 0.1,
31
+ "seq_classif_dropout": 0.2,
32
+ "sinusoidal_pos_embds": false,
33
+ "tie_weights_": true,
34
+ "torch_dtype": "float32",
35
+ "transformers_version": "4.49.0",
36
+ "vocab_size": 119547
37
+ }
onnx/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:759d7635417f923112ad7dd62fd8292bd6e909fd5bbdb9f57d5fc2edcd9797f8
3
+ size 541444865