def test_download(): # We download path1 = hub_utils.cached_download("mpariente/ConvTasNet_WHAM!_sepclean") assert os.path.isfile(path1) # We use cache path2 = hub_utils.cached_download("mpariente/ConvTasNet_WHAM!_sepclean") assert path1 == path2
def test_hf_download(model_id): # We download path1 = hub_utils.cached_download(model_id) assert os.path.isfile(path1) # We use cache path2 = hub_utils.cached_download(model_id) assert path1 == path2 # However if specifying a particular commit, # file will be different. path3 = hub_utils.cached_download( f"{HF_EXAMPLE_MODEL_IDENTIFER}@{REVISION_ID_ONE_SPECIFIC_COMMIT}") assert path3 != path1