Commit
·
6d1329a
1
Parent(s):
ff09f56
Update src/display_models/model_metadata_type.py
Browse files
src/display_models/model_metadata_type.py
CHANGED
|
@@ -21,6 +21,7 @@ class ModelType(Enum):
|
|
| 21 |
|
| 22 |
|
| 23 |
MODEL_TYPE_METADATA: Dict[str, ModelType] = {
|
|
|
|
| 24 |
"Qwen/Qwen-7B": ModelType.PT,
|
| 25 |
"Qwen/Qwen-7B-Chat": ModelType.RL,
|
| 26 |
"notstoic/PygmalionCoT-7b": ModelType.IFT,
|
|
|
|
| 21 |
|
| 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,
|