File size: 890 Bytes
649d6d4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
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')