kantii commited on
Commit
ab1c679
1 Parent(s): 673c406

Bug fix image_size as dict

Browse files

Model wasnt loading when using encoder.image_size as dict, the VisionEncoderDecoderModel class expects that is an array instead of a dictionary.

Files changed (1) hide show
  1. config.json +1 -4
config.json CHANGED
@@ -123,10 +123,7 @@
123
  "0": "LABEL_0",
124
  "1": "LABEL_1"
125
  },
126
- "image_size": {
127
- "height": 2560,
128
- "width": 1920
129
- },
130
  "initializer_range": 0.02,
131
  "is_decoder": false,
132
  "is_encoder_decoder": false,
 
123
  "0": "LABEL_0",
124
  "1": "LABEL_1"
125
  },
126
+ "image_size": [2560, 1920],
 
 
 
127
  "initializer_range": 0.02,
128
  "is_decoder": false,
129
  "is_encoder_decoder": false,