Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
shigeki Ishida
commited on
Commit
·
3cb0a6a
1
Parent(s):
ced7bdd
Add vllm version implementation
Browse files- src/display/utils.py +0 -11
src/display/utils.py
CHANGED
|
@@ -4,7 +4,6 @@ from enum import Enum
|
|
| 4 |
import pandas as pd
|
| 5 |
|
| 6 |
from src.about import Tasks, TaskType
|
| 7 |
-
from src.envs import VLLM_CURRENT_VERSION
|
| 8 |
|
| 9 |
|
| 10 |
def fields(raw_class):
|
|
@@ -171,16 +170,6 @@ class Backend(Enum):
|
|
| 171 |
return Backend.Unknown
|
| 172 |
|
| 173 |
|
| 174 |
-
class VllmVersion(Enum):
|
| 175 |
-
current = ModelDetails(VLLM_CURRENT_VERSION)
|
| 176 |
-
Unknown = ModelDetails("?")
|
| 177 |
-
|
| 178 |
-
def from_str(version):
|
| 179 |
-
if version == VLLM_CURRENT_VERSION:
|
| 180 |
-
return VllmVersion.current
|
| 181 |
-
return VllmVersion.Unknown
|
| 182 |
-
|
| 183 |
-
|
| 184 |
# Column selection
|
| 185 |
COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
|
| 186 |
TYPES = [c.type for c in fields(AutoEvalColumn)]
|
|
|
|
| 4 |
import pandas as pd
|
| 5 |
|
| 6 |
from src.about import Tasks, TaskType
|
|
|
|
| 7 |
|
| 8 |
|
| 9 |
def fields(raw_class):
|
|
|
|
| 170 |
return Backend.Unknown
|
| 171 |
|
| 172 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 173 |
# Column selection
|
| 174 |
COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
|
| 175 |
TYPES = [c.type for c in fields(AutoEvalColumn)]
|