28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
# These codes are copied from modelscope revision c58451baead80d83281f063d12fb377fad415257
|
|
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
# Cache location
|
|
from modelscope.hub.constants import DEFAULT_MODELSCOPE_DATA_ENDPOINT
|
|
|
|
DEFAULT_CACHE_HOME = Path.home().joinpath('.cache')
|
|
CACHE_HOME = os.getenv('CACHE_HOME', DEFAULT_CACHE_HOME)
|
|
DEFAULT_MS_CACHE_HOME = os.path.join(CACHE_HOME, 'modelscope', 'hub')
|
|
MS_CACHE_HOME = os.path.expanduser(
|
|
os.getenv('MS_CACHE_HOME', DEFAULT_MS_CACHE_HOME))
|
|
|
|
DEFAULT_MS_DATASETS_CACHE = os.path.join(MS_CACHE_HOME, 'datasets')
|
|
MS_DATASETS_CACHE = Path(
|
|
os.getenv('MS_DATASETS_CACHE', DEFAULT_MS_DATASETS_CACHE))
|
|
|
|
DOWNLOADED_DATASETS_DIR = 'downloads'
|
|
DEFAULT_DOWNLOADED_DATASETS_PATH = os.path.join(MS_DATASETS_CACHE,
|
|
DOWNLOADED_DATASETS_DIR)
|
|
DOWNLOADED_DATASETS_PATH = Path(
|
|
os.getenv('DOWNLOADED_DATASETS_PATH', DEFAULT_DOWNLOADED_DATASETS_PATH))
|
|
|
|
HUB_DATASET_ENDPOINT = os.environ.get('HUB_DATASET_ENDPOINT',
|
|
DEFAULT_MODELSCOPE_DATA_ENDPOINT)
|