jacoballessio commited on
Commit
dff6821
·
verified ·
1 Parent(s): ae91eeb

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +2 -11
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,16 +1,7 @@
1
  {
2
- "_name_or_path": "jacoballessio/ai-image-detect-distilled",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
6
- "id2label": {
7
- "0": "fake",
8
- "1": "real"
9
- },
10
- "label2id": {
11
- "fake": 0,
12
- "real": 1
13
- },
14
  "attention_probs_dropout_prob": 0.0,
15
  "encoder_stride": 16,
16
  "hidden_act": "gelu",
@@ -23,8 +14,8 @@
23
  "model_type": "vit",
24
  "num_attention_heads": 6,
25
  "num_channels": 3,
26
- "num_hidden_layers": 6,
27
- "patch_size": 32,
28
  "qkv_bias": true,
29
  "torch_dtype": "float32",
30
  "transformers_version": "4.41.2"
 
1
  {
 
2
  "architectures": [
3
  "ViTForImageClassification"
4
  ],
 
 
 
 
 
 
 
 
5
  "attention_probs_dropout_prob": 0.0,
6
  "encoder_stride": 16,
7
  "hidden_act": "gelu",
 
14
  "model_type": "vit",
15
  "num_attention_heads": 6,
16
  "num_channels": 3,
17
+ "num_hidden_layers": 8,
18
+ "patch_size": 16,
19
  "qkv_bias": true,
20
  "torch_dtype": "float32",
21
  "transformers_version": "4.41.2"
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c1eea1a3ad407c3e59924ecb24e62398bd6aea51d4a558b5e097fe3e7c23df90
3
- size 47403560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fe298084c3e28f12da2522290af8c48792d4f0e5cb4f70ff69bda011f27ea2b
3
+ size 58289792