File size: 779 Bytes
e94400c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
def get_vlm_model(config):
vlm_name = config.framework.qwenvl.base_vlm
if "Qwen2.5-VL" in vlm_name or "nora" in vlm_name.lower(): # temp for some ckpt
from .QWen2_5 import _QWen_VL_Interface
return _QWen_VL_Interface(config)
elif "Qwen3-VL" in vlm_name:
from .QWen3 import _QWen3_VL_Interface
return _QWen3_VL_Interface(config)
elif "florence" in vlm_name.lower(): # temp for some ckpt
from .Florence2 import _Florence_Interface
return _Florence_Interface(config)
elif "cosmos-reason2" in vlm_name.lower():
from .CosmosReason2 import _CosmosReason2_Interface
return _CosmosReason2_Interface(config)
else:
raise NotImplementedError(f"VLM model {vlm_name} not implemented")
|