Example #1
0
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
Example #2
0
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