IELTS8 commited on
Commit
78858f7
1 Parent(s): 2bbbebc

Upload model_worker.py

Browse files
Files changed (1) hide show
  1. llava/serve/model_worker.py +1 -1
llava/serve/model_worker.py CHANGED
@@ -64,7 +64,7 @@ class ModelWorker:
64
  logger.info(f"Loading the model {self.model_name} on worker {worker_id} ...")
65
  self.tokenizer, self.model, self.image_processor, self.context_len = load_pretrained_model(
66
  model_path, model_base, self.model_name, load_8bit, load_4bit, device=self.device)
67
- self.is_multimodal = 'llava' in self.model_name.lower()
68
 
69
  if not no_register:
70
  self.register_to_controller()
 
64
  logger.info(f"Loading the model {self.model_name} on worker {worker_id} ...")
65
  self.tokenizer, self.model, self.image_processor, self.context_len = load_pretrained_model(
66
  model_path, model_base, self.model_name, load_8bit, load_4bit, device=self.device)
67
+ self.is_multimodal = 'llava' or 'mavila' in self.model_name.lower()
68
 
69
  if not no_register:
70
  self.register_to_controller()