Spaces:
Sleeping
Sleeping
Upload model_worker.py
Browse files
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()
|