haiefff
commited on
Commit
•
4b6d51e
1
Parent(s):
9eb71d6
add onnx
Browse files- config.json +2 -3
- onnx/model.onnx +3 -0
- preprocessor_config.json +1 -1
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"_num_labels": 2,
|
4 |
"architectures": [
|
5 |
"ViTForImageClassification"
|
@@ -28,6 +28,5 @@
|
|
28 |
"patch_size": 16,
|
29 |
"problem_type": "single_label_classification",
|
30 |
"qkv_bias": true,
|
31 |
-
"
|
32 |
-
"transformers_version": "4.42.2"
|
33 |
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "haiefff/nsfw-images-2",
|
3 |
"_num_labels": 2,
|
4 |
"architectures": [
|
5 |
"ViTForImageClassification"
|
|
|
28 |
"patch_size": 16,
|
29 |
"problem_type": "single_label_classification",
|
30 |
"qkv_bias": true,
|
31 |
+
"transformers_version": "4.33.2"
|
|
|
32 |
}
|
onnx/model.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1611d857495dcc8e87893df5e3709790aabd17756cfc257230d1112f54dfccee
|
3 |
+
size 343463896
|
preprocessor_config.json
CHANGED
@@ -7,7 +7,7 @@
|
|
7 |
0.5,
|
8 |
0.5
|
9 |
],
|
10 |
-
"image_processor_type": "
|
11 |
"image_std": [
|
12 |
0.5,
|
13 |
0.5,
|
|
|
7 |
0.5,
|
8 |
0.5
|
9 |
],
|
10 |
+
"image_processor_type": "ViTFeatureExtractor",
|
11 |
"image_std": [
|
12 |
0.5,
|
13 |
0.5,
|