INNER CODE UNIT · Python

get_vqascore_model

linzhiqiu/t2v_metrics · t2v_metrics/models/vqascore_models/__init__.py:31

def get_vqascore_model(model_name, device='cuda', cache_dir=HF_CACHE_DIR, **kwargs):
    assert model_name in list_all_vqascore_models()
    if model_name in GPT4V_MODELS:
       return GPT4VModel(model_name, device=device, cache_dir=cache_dir, **kwargs)
    elif model_name in PALIGEMMA_MODELS:
        return PaliGemmaModel(model_name, device=device, cache_dir=cache_dir, **kwargs)
    # elif model_name in INTERNVL_MODELS:
    #     return InternVLModel(model_name, device=device, cache_dir=cache_dir, **kwargs)
    elif model_name in GEMINI_MODELS:
        return GeminiModel(model_name, device=device, cache_dir=cache_dir, **kwargs)
    elif model_name in QWEN2_VL_MODELS:
        return Qwen2VLModel(model_name, device=device, cache_dir=cache_dir, **kwargs)
    elif model_name in QWEN3_VL_MODELS:
        return Qwen3VLModel(model_name, device=device, cache_dir=cache_dir, **kwargs)
    elif model_name in QWEN3_OMNI_MODELS:
        return Qwen3OmniModel(model_name, device=device, cache_dir=cache_dir, **kwargs)
    elif model_name in GEMMA3_MODELS:
        return Gemma3Model(model_name, device=device, cache_dir=cache_dir, **kwargs)

View source record →