INNER CODE UNIT · Python

list_all_models

linzhiqiu/t2v_metrics · t2v_metrics/__init__.py:26

def list_all_models():
    return list_all_vqascore_models() 

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 →