Uzef commited on
Commit
48e8a2f
·
verified ·
1 Parent(s): 229d316

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +32 -0
config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "DEVICE": "cpu",
3
+ "_name_or_path": "not-lain/deepfake",
4
+ "architectures": [
5
+ "DeepFakeModel"
6
+ ],
7
+ "auto_map": {
8
+ "AutoConfig": "deepfakeconfig.DeepFakeConfig",
9
+ "AutoModelForImageClassification": "deepfakemodel.DeepFakeModel"
10
+ },
11
+ "custom_pipelines": {
12
+ "deepfake": {
13
+ "default": {
14
+ "model": {
15
+ "pt": [
16
+ "not-lain/deepfake",
17
+ "main"
18
+ ]
19
+ }
20
+ },
21
+ "impl": "pipeline.DeepFakePipeline",
22
+ "pt": [
23
+ "AutoModelForImageClassification"
24
+ ],
25
+ "tf": [],
26
+ "type": "multimodal"
27
+ }
28
+ },
29
+ "model_type": "ResNet",
30
+ "torch_dtype": "float32",
31
+ "transformers_version": "4.26.1"
32
+ }