jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix_smallbugs_bbeh

Browse Source
This commit is contained in:
yufeng zhao 2025-03-10 04:54:28 +00:00
parent 28ad5a98d4
commit 5fc26c92b2
3 changed files with 2 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
hf_settings.py
View File

@ -1,23 +0,0 @@
import os
import huggingface_hub.constants as hf_constants
from huggingface_hub import set_cache_dir
from datasets import get_dataset_config_names # Optional, if you need dataset-related functionality
# Set a new cache directory
new_cache_dir = "/fs-computility/llm/shared/llmeval/models/opencompass_hf_hub" # Replace with your desired path
set_cache_dir(new_cache_dir)
# Alternatively, you can set the environment variable
# os.environ["HF_HOME"] = new_cache_dir
# Root cache path for Hugging Face
root_cache_dir = hf_constants.HF_HOME
print(f"Root HF cache path: {root_cache_dir}")
# Dataset cache path (typically under HF_HOME/datasets)
dataset_cache_dir = f"{root_cache_dir}/datasets"
print(f"Dataset cache path: {dataset_cache_dir}")
# Model cache path (typically under HF_HOME/hub)
model_cache_dir = f"{root_cache_dir}/hub"
print(f"Model cache path: {model_cache_dir}")

1
opencompass/datasets/__init__.py
View File

@ -12,6 +12,7 @@ from .babilong import * # noqa: F401, F403
from .bbeh import * # noqa: F401, F403
from .bbh import * # noqa: F401, F403
from .bbeh import * # noqa: F401, F403
from .bbh import * # noqa: F401, F403
from .bigcodebench import * # noqa: F401, F403
from .boolq import * # noqa: F401, F403
from .bustum import * # noqa: F401, F403