Clémentine commited on
Commit
1d49827
1 Parent(s): 512b095

type updated

Browse files
src/display_models/model_metadata_type.py CHANGED
@@ -22,6 +22,8 @@ class ModelType(Enum):
22
 
23
  MODEL_TYPE_METADATA: Dict[str, ModelType] = {
24
  "tiiuae/falcon-180B": ModelType.PT,
 
 
25
  "Qwen/Qwen-7B": ModelType.PT,
26
  "Qwen/Qwen-7B-Chat": ModelType.RL,
27
  "notstoic/PygmalionCoT-7b": ModelType.IFT,
 
22
 
23
  MODEL_TYPE_METADATA: Dict[str, ModelType] = {
24
  "tiiuae/falcon-180B": ModelType.PT,
25
+ "tiiuae/falcon-180B-chat": ModelType.RL,
26
+ "microsoft/phi-1_5": ModelType.PT,
27
  "Qwen/Qwen-7B": ModelType.PT,
28
  "Qwen/Qwen-7B-Chat": ModelType.RL,
29
  "notstoic/PygmalionCoT-7b": ModelType.IFT,