获取实际模型名称,处理多模态模型 Args: source_name (str): 源名称 model_name (str): 模型名称 Returns: str: 实际模型名称或None
(self, source_name, model_name)
| 297 | raise |
| 298 | |
| 299 | def _get_actual_model_name(self, source_name, model_name): |
| 300 | """获取实际模型名称,处理多模态模型 |
| 301 | |
| 302 | Args: |
| 303 | source_name (str): 源名称 |
| 304 | model_name (str): 模型名称 |
| 305 | |
| 306 | Returns: |
| 307 | str: 实际模型名称或None |
| 308 | """ |
| 309 | # 获取基础模型名(去掉_mm后缀) |
| 310 | base_model_name = model_name[:-3] if model_name.endswith("_mm") else model_name |
| 311 | |
| 312 | if not (source_name in self.source_mapping and base_model_name in self.source_mapping[source_name]): |
| 313 | return None |
| 314 | |
| 315 | # 如果模型映射值为None且是多模态模型(以_mm结尾) |
| 316 | if self.source_mapping[source_name][base_model_name] is None and model_name.endswith("_mm"): |
| 317 | # 多模态模型映射为None表示不支持 |
| 318 | return None |
| 319 | |
| 320 | return self.source_mapping[source_name][base_model_name] |
| 321 | |
| 322 | def get_config(self, model_name, mode, input_proportion: int, output_proportion: int): |
| 323 | """维护两套策略, 一套是以便宜为导向, |
no outgoing calls
no test coverage detected