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}"
|