Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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,
|