From 77fa31aeb84945efdc1f0f25025ece6faee7259d Mon Sep 17 00:00:00 2001
From: Niels Rogge <niels.rogge1@gmail.com>
Date: Fri, 15 Oct 2021 08:42:13 +0000
Subject: [PATCH] Update model

---
 config.json              | 5 +++--
 preprocessor_config.json | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/config.json b/config.json
index bdae24c..8c022d8 100644
--- a/config.json
+++ b/config.json
@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "SegFormerForImageSegmentation"
+    "SegformerForImageSegmentation"
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
@@ -354,6 +354,7 @@
     3,
     3
   ],
+  "reshape_last_stage": true,
   "sr_ratios": [
     8,
     4,
@@ -367,5 +368,5 @@
     2
   ],
   "torch_dtype": "float32",
-  "transformers_version": "4.10.0.dev0"
+  "transformers_version": "4.12.0.dev0"
 }
diff --git a/preprocessor_config.json b/preprocessor_config.json
index 0ee8471..f9ec06e 100644
--- a/preprocessor_config.json
+++ b/preprocessor_config.json
@@ -8,7 +8,7 @@
   "do_pad": true,
   "do_random_crop": false,
   "do_resize": true,
-  "feature_extractor_type": "SegFormerFeatureExtractor",
+  "feature_extractor_type": "SegformerFeatureExtractor",
   "image_mean": [
     0.485,
     0.456,