19 lines
492 B
Python
19 lines
492 B
Python
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
|