Update chat.py

chat.py

@@ -34,7 +34,7 @@ class Chat:
         self.tokenizer, self.model, self.image_processor, self.seg_image_processor, self.depth_image_processor, self.context_len = load_pretrained_model(
             model_path, model_base, self.model_name, load_8bit, load_4bit, device=self.device)
         self.is_multimodal = 'llava' in self.model_name.lower()
-        self.is_seg = "seg_llava" in self.model_name.lower()
+        self.is_seg = "vcoder" in self.model_name.lower()
         self.is_depth = False
 
     @torch.inference_mode()
@@ -202,4 +202,4 @@ if __name__ == "__main__":
     parser.add_argument("--no-register", action="store_true")
     parser.add_argument("--load-8bit", action="store_true")
     parser.add_argument("--load-4bit", action="store_true")
-    args = parser.parse_args()
+    args = parser.parse_args()