Signed-off-by: bk-201 <joy25810@foxmail.com>
This commit is contained in:
bk-201 2025-10-13 02:41:29 +00:00
parent 882213cca2
commit 8863bd2b74

View File

@ -390,12 +390,12 @@ class LoRAModelManager:
) )
else: else:
self.supports_mm_lora = False self.supports_mm_lora = False
if self.supports_mm_lora: # 从init传进来就可以了不需要model_config了 if self.supports_mm_lora:
self.mm_mapping: MultiModelKeys = self.model.get_mm_mapping() self.mm_mapping: MultiModelKeys = self.model.get_mm_mapping()
self.mm_config = model_config.multimodal_config self.mm_config = model_config.multimodal_config
# limit_per_prompt: int = max( # limit_per_prompt: int = max(
# self.info.get_allowed_mm_limits().values()) # self.info.get_allowed_mm_limits().values())
limit_per_prompt = 5 limit_per_prompt = 5 # TODO
# For vision tower # For vision tower
# max_num_batched_tokens = encoder_budget # max_num_batched_tokens = encoder_budget