INNER CODE UNIT · Python

get_score_model

linzhiqiu/t2v_metrics · t2v_metrics/__init__.py:29

def get_score_model(model='clip-flant5-xxl', device='cuda', cache_dir=HF_CACHE_DIR, **kwargs):
    if model in list_all_vqascore_models():
        return VQAScore(model, device=device, cache_dir=cache_dir, **kwargs)
    else:
        raise NotImplementedError()

View source record →