opensora/utils/downloader.py

import gdown
import os

opensora_cache_home = os.path.expanduser(
    os.getenv("OPENSORA_HOME", os.path.join("~/.cache", "opensora"))
)


def gdown_download(id, fname, cache_dir=None):
    cache_dir = opensora_cache_home if not cache_dir else cache_dir

    os.makedirs(cache_dir, exist_ok=True)
    destination = os.path.join(cache_dir, fname)
    if os.path.exists(destination):
        return destination

    gdown.download(id=id, output=destination, quiet=False)
    return destination