File size: 672 Bytes
54f3112 |
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 dataclasses import dataclass
from enum import Enum
import glob
import json
import os
from typing import Dict, List
from ..utils_display import AutoEvalColumn
@dataclass
class ModelInfo:
name: str
symbol: str # emoji
model_type_symbols = {
"LLM": "🟢",
"ImageLLM": "🔶",
"VideoLLM": "⭕",
"Other": "🟦",
}
class ModelType(Enum):
PT = ModelInfo(name="LLM", symbol="🟢")
FT = ModelInfo(name="ImageLLM", symbol="🔶")
IFT = ModelInfo(name="VideoLLM", symbol="⭕")
RL = ModelInfo(name="Other", symbol="🟦")
def to_str(self, separator = " "):
return f"{self.value.symbol}{separator}{self.value.name}"
|