Update model_UI.py

model_UI.py

@@ -4,10 +4,20 @@ import os
 import json
 from tqdm import tqdm
 
-from constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN, DEFAULT_REGION_FEA_TOKEN, VOCAB_IMAGE_W, VOCAB_IMAGE_H
+IMAGE_TOKEN_INDEX = -200
+DEFAULT_IMAGE_TOKEN = "<image>"
+DEFAULT_IMAGE_PATCH_TOKEN = "<im_patch>"
+DEFAULT_IM_START_TOKEN = "<im_start>"
+DEFAULT_IM_END_TOKEN = "<im_end>"
+IMAGE_PLACEHOLDER = "<image-placeholder>"
+
+# Added by Ferret
+DEFAULT_REGION_FEA_TOKEN = "<region_fea>"
+VOCAB_IMAGE_W = 1000
+VOCAB_IMAGE_H = 1000
 from conversation import conv_templates, SeparatorStyle
 from builder import load_pretrained_model
-from utils import disable_torch_init
+
 from mm_utils import tokenizer_image_token, process_images
 
 from PIL import Image
@@ -17,6 +27,13 @@ import numpy as np
 from copy import deepcopy
 from functools import partial
 
+def disable_torch_init():
+    """
+    Disable the redundant torch default initialization to accelerate model creation.
+    """
+    import torch
+    setattr(torch.nn.Linear, "reset_parameters", lambda self: None)
+    setattr(torch.nn.LayerNorm, "reset_parameters", lambda self: None)
 
 def split_list(lst, n):
     """Split a list into n (roughly) equal-sized chunks"""