Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
from src.envs import ( | |
API, | |
EVAL_REQUESTS_PATH, | |
DYNAMIC_INFO_REPO, | |
DYNAMIC_INFO_FILE_PATH, | |
DYNAMIC_INFO_PATH, | |
EVAL_RESULTS_PATH, | |
H4_TOKEN, IS_PUBLIC, | |
QUEUE_REPO, | |
REPO_ID, | |
RESULTS_REPO, | |
SHOW_INCOMPLETE_EVALS | |
) | |
from huggingface_hub import snapshot_download | |
import os | |
import glob | |
import json | |
snapshot_download(repo_id=QUEUE_REPO, local_dir=EVAL_REQUESTS_PATH, repo_type="dataset", tqdm_class=None, etag_timeout=30) | |
all_models = [] | |
for filepath in glob.glob(os.path.join(EVAL_REQUESTS_PATH, '**/*.json'), recursive=True): | |
with open(filepath, 'r') as f: | |
model_data = json.load(f) | |
if model_data['status'] == 'FINISHED': | |
all_models.append(model_data['model']) | |
all_models = sorted(list(set(all_models))) | |
with open('model_list.txt', 'w') as fw: | |
for m in all_models: | |
print(m) | |
fw.write(' - '+ m + '\n') |