diff --git a/python/sglang/srt/lora/lora_manager.py b/python/sglang/srt/lora/lora_manager.py index baf120ca2..d288323bf 100644 --- a/python/sglang/srt/lora/lora_manager.py +++ b/python/sglang/srt/lora/lora_manager.py @@ -415,6 +415,10 @@ class LoRAManager: replace_submodule(self.base_model, module_name, lora_module) return lora_module + def should_skip_lora_for_vision_model(self, module_name): + # TODO: support different vision models + return module_name.find("vision_model.model") != -1 + def init_lora_modules(self): # Look-up table that essentially maps (layer_index, module_name) to the corresponding LoRA module. self.lora_modules: List[Dict[str, BaseLayerWithLoRA]] = [ @@ -432,6 +436,10 @@ class LoRAManager: ) and not self.base_model.should_apply_lora(module_name): continue + # Skip vision model + if self.should_skip_lora_for_vision_model(module_name): + continue + # The module should be converted if it is included in target_names if module_name.split(".")[-1] in self.target_modules: layer_id = get_layer_id(module_name)