FM / huggingface

/usr/local/lib/python3.6/site-packages/huggingface_hub/

/usr/local/lib/python3.6/site-packages/huggingface_hub

Name	Size	Mode	Actions
commands/	-	0755	rm
utils/	-	0755	rm
__pycache__/	-	0755	rm
constants.py	1645	0644	edit dl rm
file_download.py	18847	0644	edit dl rm
hf_api.py	54138	0644	edit dl rm
hub_mixin.py	15329	0644	edit dl rm
inference_api.py	5445	0644	edit dl rm
keras_mixin.py	9124	0644	edit dl rm
lfs.py	1426	0644	edit dl rm
repocard.py	3197	0644	edit dl rm
repocard_types.py	1522	0644	edit dl rm
repository.py	49785	0644	edit dl rm
snapshot_download.py	7974	0644	edit dl rm
__init__.py	2023	0644	edit dl rm

Edit: /usr/local/lib/python3.6/site-packages/huggingface_hub/constants.py (1645B)

import os

# Possible values for env variables

ENV_VARS_TRUE_VALUES = {"1", "ON", "YES", "TRUE"}
ENV_VARS_TRUE_AND_AUTO_VALUES = ENV_VARS_TRUE_VALUES.union({"AUTO"})

# Constants for file downloads

PYTORCH_WEIGHTS_NAME = "pytorch_model.bin"
TF2_WEIGHTS_NAME = "tf_model.h5"
TF_WEIGHTS_NAME = "model.ckpt"
FLAX_WEIGHTS_NAME = "flax_model.msgpack"
CONFIG_NAME = "config.json"
REPOCARD_NAME = "README.md"

DEFAULT_REVISION = "main"

HUGGINGFACE_CO_URL_HOME = "https://huggingface.co/"

ENV_VARS_TRUE_VALUES = {"1", "ON", "YES", "TRUE"}
_staging_mode = (
    os.environ.get("HUGGINGFACE_CO_STAGING", "NO").upper() in ENV_VARS_TRUE_VALUES
)

ENDPOINT = (
    "https://moon-staging.huggingface.co" if _staging_mode else "https://huggingface.co"
)

HUGGINGFACE_CO_URL_TEMPLATE = ENDPOINT + "/{repo_id}/resolve/{revision}/{filename}"

REPO_TYPE_DATASET = "dataset"
REPO_TYPE_SPACE = "space"
REPO_TYPES = [None, REPO_TYPE_DATASET, REPO_TYPE_SPACE]
SPACES_SDK_TYPES = ["gradio", "streamlit", "static"]

REPO_TYPES_URL_PREFIXES = {
    REPO_TYPE_DATASET: "datasets/",
    REPO_TYPE_SPACE: "spaces/",
}
REPO_TYPES_MAPPING = {"datasets": REPO_TYPE_DATASET, "spaces": REPO_TYPE_SPACE}

# default cache
hf_cache_home = os.path.expanduser(
    os.getenv(
        "HF_HOME", os.path.join(os.getenv("XDG_CACHE_HOME", "~/.cache"), "huggingface")
    )
)
default_cache_path = os.path.join(hf_cache_home, "hub")

HUGGINGFACE_HUB_CACHE = os.getenv("HUGGINGFACE_HUB_CACHE", default_cache_path)

HF_HUB_OFFLINE = os.environ.get("HF_HUB_OFFLINE", "AUTO").upper()
if HF_HUB_OFFLINE in ENV_VARS_TRUE_VALUES:
    HF_HUB_OFFLINE = True
else:
    HF_HUB_OFFLINE = False

import os # Possible values for env variables ENV_VARS_TRUE_VALUES = {"1", "ON", "YES", "TRUE"} ENV_VARS_TRUE_AND_AUTO_VALUES = ENV_VARS_TRUE_VALUES.union({"AUTO"}) # Constants for file downloads PYTORCH_WEIGHTS_NAME = "pytorch_model.bin" TF2_WEIGHTS_NAME = "tf_model.h5" TF_WEIGHTS_NAME = "model.ckpt" FLAX_WEIGHTS_NAME = "flax_model.msgpack" CONFIG_NAME = "config.json" REPOCARD_NAME = "README.md" DEFAULT_REVISION = "main" HUGGINGFACE_CO_URL_HOME = "https://huggingface.co/" ENV_VARS_TRUE_VALUES = {"1", "ON", "YES", "TRUE"} _staging_mode = ( os.environ.get("HUGGINGFACE_CO_STAGING", "NO").upper() in ENV_VARS_TRUE_VALUES ) ENDPOINT = ( "https://moon-staging.huggingface.co" if _staging_mode else "https://huggingface.co" ) HUGGINGFACE_CO_URL_TEMPLATE = ENDPOINT + "/{repo_id}/resolve/{revision}/{filename}" REPO_TYPE_DATASET = "dataset" REPO_TYPE_SPACE = "space" REPO_TYPES = [None, REPO_TYPE_DATASET, REPO_TYPE_SPACE] SPACES_SDK_TYPES = ["gradio", "streamlit", "static"] REPO_TYPES_URL_PREFIXES = { REPO_TYPE_DATASET: "datasets/", REPO_TYPE_SPACE: "spaces/", } REPO_TYPES_MAPPING = {"datasets": REPO_TYPE_DATASET, "spaces": REPO_TYPE_SPACE} # default cache hf_cache_home = os.path.expanduser( os.getenv( "HF_HOME", os.path.join(os.getenv("XDG_CACHE_HOME", "~/.cache"), "huggingface") ) ) default_cache_path = os.path.join(hf_cache_home, "hub") HUGGINGFACE_HUB_CACHE = os.getenv("HUGGINGFACE_HUB_CACHE", default_cache_path) HF_HUB_OFFLINE = os.environ.get("HF_HUB_OFFLINE", "AUTO").upper() if HF_HUB_OFFLINE in ENV_VARS_TRUE_VALUES: HF_HUB_OFFLINE = True else: HF_HUB_OFFLINE = False