FredZhang7 commited on
Commit
1f67134
·
1 Parent(s): 2370f1b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -0
config.json CHANGED
@@ -1,4 +1,9 @@
1
  {
 
 
 
 
 
2
  "model_type": "CustomEfficientNetV2",
3
  "url": "https://huggingface.co/FredZhang7/efficientnetv2.5_rw_s/resolve/main/efficientnetv2.5_base_in1k.pth",
4
  "input_size": [
 
1
  {
2
+ "_from_model_config": true,
3
+ "auto_map": {
4
+ "AutoConfig": "custom_config.CustomEfficientNetV2Config",
5
+ "AutoModel": "custom_efficientnetv2.CustomEfficientNetV2"
6
+ }
7
  "model_type": "CustomEfficientNetV2",
8
  "url": "https://huggingface.co/FredZhang7/efficientnetv2.5_rw_s/resolve/main/efficientnetv2.5_base_in1k.pth",
9
  "input_size": [