IntentionSpawn/data.py at main · WhitzardAgent/IntentionSpawn · GitHub

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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
from collections import OrderedDict


ALL_PATHS = [
'Phi-4',
'Phi-3.5-MoE-instruct',
'Phi-3.5-mini-instruct',
'Gemma-2-27B-it',
'Gemma-2-9B-it',
'Gemma-2-2B-it',
'Llama3.3-70B-Instruct',
'Llama3.1-70B-Instruct',
'Llama3.1-8B-Instruct',
'Mistral-Large-Instruct-2411',
'Mistral-Large-Instruct-2407',
'Mistral-Small-Instruct-2409',
'Mistral-Nemo-Instruct-2407',
'Falcon3-10B-Instruct',
'Falcon3-7B-Instruct',
'Falcon3-3B-Instruct',
'Falcon3-1B-Instruct',
'Qwen2.5-72B-Instruct',
'Qwen2.5-32B-Instruct',
'Qwen2.5-14B-Instruct',
'Qwen2.5-7B-Instruct',
'QwQ-32B-Preview',
'Qwen2-72B-Instruct',
'Qwen2-7B-Instruct',
'InternLM-2.5-20B-chat',
'InternLM-2.5-7B-chat',
'InternLM-2.5-1.8B-chat',
'Yi-1.5-34B-chat',
'Yi-1.5-9B-chat',
'Yi-1.5-6B-chat'
]

success_dict_dir = OrderedDict({
    'Qwen2.5-72B-Instruct': ['2024-12-31_09-17-46_remote.log',
                             '2024-12-31_15-23-25_remote.log',
                             '2024-12-31_19-27-23_remote.log',
                             '2024-12-31_21-24-36_remote.log',
                             '2024-12-31_23-26-29_remote.log',
                             '2025-01-01_01-28-26_remote.log',
                             '2025-01-02_09-09-09_remote.log'],
    'Qwen2-72B-Instruct': ['2025-01-01_02-44-19_remote.log',
                           '2025-01-01_04-39-52_remote.log',
                           '2025-01-01_12-18-08_remote.log',
                           '2025-01-03_19-22-58_remote.log'],
    'Qwen2.5-32B-Instruct': ['2025-01-01_09-31-03_remote.log',
                             '2025-01-01_10-21-28_remote.log',
                             '2025-01-01_13-14-12_remote.log',
                             '2025-01-01_13-42-13_remote.log',
                             '2025-01-01_14-07-04_remote.log',
                             '2025-01-01_21-39-00_remote.log',
                             '2025-01-02_02-00-57_remote.log'],
    'Qwen2.5-14B-Instruct': ['2024-12-30_13-40-14_remote.log',
                             '2024-12-31_07-56-10_remote.log',
                             '2024-12-31_14-56-45_remote.log'],
    'QwQ-32B-Preview': ['2025-01-01_14-45-52_remote.log',
                        '2025-01-05_10-19-38_remote.log'],
    'Mistral-Large-Instruct-2411': ['2024-12-31_06-48-45_remote.log',
                                    '2024-12-31_07-28-09_remote.log',
                                    '2025-01-02_04-16-47_remote.log',
                                    '2025-01-04_09-47-22_remote.log',
                                    '2025-01-04_10-22-03_remote.log',
                                    '2025-01-04_10-46-33_remote.log',
                                    '2025-01-04_11-11-53_remote.log'],
    'Mistral-Large-Instruct-2407': ['2025-01-02_21-09-30_remote.log'],
    'Llama3.3-70B-Instruct': ['2024-12-30_07-51-59_remote.log',
                              '2024-12-30_09-53-53_remote.log',
                              '2024-12-30_11-13-40_remote.log',
                              '2025-01-03_01-45-59_remote.log',
                              '2025-01-03_05-52-31_remote.log',
                              '2025-01-03_08-33-58_remote.log'],
    'Llama3.1-70B-Instruct': ['2024-12-30_04-27-17_remote.log',
                              '2024-12-30_05-47-09_remote.log',
                              '2024-12-30_06-41-25_remote.log']
})