ZhengPeng7 commited on
Commit
1c4ea53
1 Parent(s): fc65c0f

For users to load in one key.

Browse files
Files changed (1) hide show
  1. config.json +3 -1
config.json CHANGED
@@ -16,5 +16,7 @@
16
  "type": "image"
17
  }
18
  },
19
- "bb_pretrained": false
 
 
20
  }
 
16
  "type": "image"
17
  }
18
  },
19
+ "bb_pretrained": false,
20
+ "model_type": "SegformerForSemanticSegmentation",
21
+ "torch_dtype": "float32",
22
  }