future-xy commited on
Commit
5fd4d0a
1 Parent(s): f8f8961
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 = "HF-Chat"
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("MoE-Infinity")
138
- HF_Chat = ModelDetails("HF-Chat")
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):