Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 1,122 Bytes
649d6d4 b4fc70b 78e9bc6 b4fc70b 649d6d4 b4fc70b 649d6d4 b4fc70b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
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 = []
with open('external_models_results.json', 'r') as f:
external_models = json.load(f)
for model in external_models:
if 'huggingface.co/' in model['link']:
all_models.append(model['link'].split('huggingface.co/')[-1])
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') |