INNER CODE UNIT · Python

list_all_vqascore_models

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

def list_all_vqascore_models():
    return [model for models in ALL_VQA_MODELS for model in models]

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:

View source record →