update important models
Browse files
data.py
CHANGED
|
@@ -20,10 +20,13 @@ IMPORTANT_MODELS = [
|
|
| 20 |
"vit", # old (vision) - fixed comma
|
| 21 |
"clip", # old but dominant (vision)
|
| 22 |
"detr", # objection detection, segmentation (vision)
|
| 23 |
-
"
|
| 24 |
"got_ocr2", # ocr (vision)
|
| 25 |
"whisper", # old but dominant (audio)
|
| 26 |
"wav2vec2", # old (audio)
|
|
|
|
|
|
|
|
|
|
| 27 |
"llama", # new and dominant (meta)
|
| 28 |
"gemma3", # new (google)
|
| 29 |
"qwen2", # new (Alibaba)
|
|
@@ -34,6 +37,8 @@ IMPORTANT_MODELS = [
|
|
| 34 |
"internvl", # new (video)
|
| 35 |
"gemma3n", # new (omnimodal models)
|
| 36 |
"qwen2_5_omni", # new (omnimodal models)
|
|
|
|
|
|
|
| 37 |
]
|
| 38 |
|
| 39 |
KEYS_TO_KEEP = [
|
|
|
|
| 20 |
"vit", # old (vision) - fixed comma
|
| 21 |
"clip", # old but dominant (vision)
|
| 22 |
"detr", # objection detection, segmentation (vision)
|
| 23 |
+
"table_transformer", # objection detection (visioin) - maybe just detr?
|
| 24 |
"got_ocr2", # ocr (vision)
|
| 25 |
"whisper", # old but dominant (audio)
|
| 26 |
"wav2vec2", # old (audio)
|
| 27 |
+
"qwen2_audio", # (audio)
|
| 28 |
+
"speech_t5", # (audio)
|
| 29 |
+
"csm", # (audio)
|
| 30 |
"llama", # new and dominant (meta)
|
| 31 |
"gemma3", # new (google)
|
| 32 |
"qwen2", # new (Alibaba)
|
|
|
|
| 37 |
"internvl", # new (video)
|
| 38 |
"gemma3n", # new (omnimodal models)
|
| 39 |
"qwen2_5_omni", # new (omnimodal models)
|
| 40 |
+
# "gpt_oss", # new (quite used)
|
| 41 |
+
"qwen2_5_omni", # new (omnimodal models)
|
| 42 |
]
|
| 43 |
|
| 44 |
KEYS_TO_KEEP = [
|