Update utils.py

utils.py

@@ -94,7 +94,7 @@ def get_transform(is_train, image_size):
     return transform
 
 def post_process(vit_embeds, target_aspect_ratio, model_type='VIT'):
-    if model_type in ["TokenOCR-4096-English-seg", "TokenOCR-2048-Bilingual-seg"]:
+    if model_type in ["TokenFD_4096_English_seg", "TokenFD_2048_Bilingual_seg"]:
         h = w = int(vit_embeds.shape[1] ** 0.5)
         c = vit_embeds.shape[-1]
         # vit_embeds_local = vit_embeds[:-1].reshape(-1, h, w, c).permute(0, 3, 1, 2)