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