| 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 | |