future-xy
commited on
Commit
•
5fd4d0a
1
Parent(s):
f8f8961
fix bug
Browse files- src/backend/manage_requests.py +1 -1
- src/display/utils.py +2 -2
src/backend/manage_requests.py
CHANGED
@@ -16,7 +16,7 @@ class EvalRequest:
|
|
16 |
json_filepath: str
|
17 |
weight_type: str = "Original"
|
18 |
model_type: str = "" # pretrained, finetuned, with RL
|
19 |
-
inference_framework: str = "
|
20 |
precision: str = "" # float16, bfloat16
|
21 |
base_model: Optional[str] = None # for adapter models
|
22 |
revision: str = "main" # commit
|
|
|
16 |
json_filepath: str
|
17 |
weight_type: str = "Original"
|
18 |
model_type: str = "" # pretrained, finetuned, with RL
|
19 |
+
inference_framework: str = "hf-chat"
|
20 |
precision: str = "" # float16, bfloat16
|
21 |
base_model: Optional[str] = None # for adapter models
|
22 |
revision: str = "main" # commit
|
src/display/utils.py
CHANGED
@@ -134,8 +134,8 @@ class ModelType(Enum):
|
|
134 |
|
135 |
class InferenceFramework(Enum):
|
136 |
# "moe-infinity", hf-chat
|
137 |
-
MoE_Infinity = ModelDetails("
|
138 |
-
HF_Chat = ModelDetails("
|
139 |
Unknown = ModelDetails("?")
|
140 |
|
141 |
def to_str(self):
|
|
|
134 |
|
135 |
class InferenceFramework(Enum):
|
136 |
# "moe-infinity", hf-chat
|
137 |
+
MoE_Infinity = ModelDetails("moe-infinity")
|
138 |
+
HF_Chat = ModelDetails("hf-chat")
|
139 |
Unknown = ModelDetails("?")
|
140 |
|
141 |
def to_str(self):
|