amcoff commited on
Commit
d1b0c69
·
unverified ·
1 Parent(s): 852edbc

retrain on new data

Browse files
Files changed (6) hide show
  1. .gitattributes +1 -0
  2. .gitignore +1 -0
  3. config.json +4 -2
  4. model.onnx +1 -1
  5. pytorch_model.bin +1 -1
  6. requirements.txt +53 -0
.gitattributes CHANGED
@@ -1,2 +1,3 @@
1
  model.onnx filter=lfs diff=lfs merge=lfs -text
2
  tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
1
  model.onnx filter=lfs diff=lfs merge=lfs -text
2
  tokenizer.json filter=lfs diff=lfs merge=lfs -text
3
+ pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .venv
config.json CHANGED
@@ -14,14 +14,14 @@
14
  "2": "ANY",
15
  "3": "SIDE",
16
  "4": "MESSAGE",
17
- "5": "FEST"
18
  },
19
  "initializer_range": 0.02,
20
  "intermediate_size": 3072,
21
  "label2id": {
22
  "ANY": 2,
23
- "FEST": 5,
24
  "FOOD": 0,
 
25
  "MESSAGE": 4,
26
  "NO": 1,
27
  "SIDE": 3
@@ -36,6 +36,8 @@
36
  "position_embedding_type": "absolute",
37
  "problem_type": "single_label_classification",
38
  "torch_dtype": "float32",
 
39
  "type_vocab_size": 2,
 
40
  "vocab_size": 50325
41
  }
 
14
  "2": "ANY",
15
  "3": "SIDE",
16
  "4": "MESSAGE",
17
+ "5": "GENERIC"
18
  },
19
  "initializer_range": 0.02,
20
  "intermediate_size": 3072,
21
  "label2id": {
22
  "ANY": 2,
 
23
  "FOOD": 0,
24
+ "GENERIC": 5,
25
  "MESSAGE": 4,
26
  "NO": 1,
27
  "SIDE": 3
 
36
  "position_embedding_type": "absolute",
37
  "problem_type": "single_label_classification",
38
  "torch_dtype": "float32",
39
+ "transformers_version": "4.26.1",
40
  "type_vocab_size": 2,
41
+ "use_cache": true,
42
  "vocab_size": 50325
43
  }
model.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:99cc4c2e41b7364950194c99815457b12055170a69531f1f4898085735e36ed7
3
  size 496686067
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51c3e497b5ab8f7a7f8dffe787b40fd642b94c3da1c263549acbf5cae8ebc493
3
  size 496686067
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8aa464ebd569021a7cf3ee03cdcad0f3dee675e19540ce9a58fe7c71d452bc5c
3
  size 498855029
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b0b98af86e28013f6685a7a47dc932e8f8039008c2968fee1e897f6eee37a23a
3
  size 498855029
requirements.txt ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ aiohttp==3.8.4
2
+ aiosignal==1.3.1
3
+ async-timeout==4.0.2
4
+ attrs==22.2.0
5
+ certifi==2022.12.7
6
+ charset-normalizer==3.0.1
7
+ coloredlogs==15.0.1
8
+ datasets==2.10.0
9
+ dill==0.3.6
10
+ evaluate==0.4.0
11
+ filelock==3.9.0
12
+ flatbuffers==23.1.21
13
+ frozenlist==1.3.3
14
+ fsspec==2023.1.0
15
+ huggingface-hub==0.12.1
16
+ humanfriendly==10.0
17
+ idna==3.4
18
+ joblib==1.2.0
19
+ mpmath==1.2.1
20
+ multidict==6.0.4
21
+ multiprocess==0.70.14
22
+ numpy==1.23.5
23
+ nvidia-cublas-cu11==11.10.3.66
24
+ nvidia-cuda-nvrtc-cu11==11.7.99
25
+ nvidia-cuda-runtime-cu11==11.7.99
26
+ nvidia-cudnn-cu11==8.5.0.96
27
+ onnx==1.12.0
28
+ onnxruntime==1.14.0
29
+ optimum==1.6.4
30
+ packaging==23.0
31
+ pandas==1.5.3
32
+ protobuf==3.20.1
33
+ pyarrow==11.0.0
34
+ python-dateutil==2.8.2
35
+ pytz==2022.7.1
36
+ PyYAML==6.0
37
+ regex==2022.10.31
38
+ requests==2.28.2
39
+ responses==0.18.0
40
+ scikit-learn==1.2.1
41
+ scipy==1.10.1
42
+ sentencepiece==0.1.97
43
+ six==1.16.0
44
+ sympy==1.11.1
45
+ threadpoolctl==3.1.0
46
+ tokenizers==0.13.2
47
+ torch==1.13.1
48
+ tqdm==4.64.1
49
+ transformers==4.26.1
50
+ typing_extensions==4.5.0
51
+ urllib3==1.26.14
52
+ xxhash==3.2.0
53
+ yarl==1.8.2