Spaces:
Runtime error
Runtime error
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,
|