Commit
•
8541f99
1
Parent(s):
3ae1b8c
Update src/display_models/model_metadata_type.py
Browse files
src/display_models/model_metadata_type.py
CHANGED
|
@@ -21,6 +21,8 @@ class ModelType(Enum):
|
|
| 21 |
|
| 22 |
|
| 23 |
MODEL_TYPE_METADATA: Dict[str, ModelType] = {
|
|
|
|
|
|
|
| 24 |
"notstoic/PygmalionCoT-7b": ModelType.IFT,
|
| 25 |
"aisquared/dlite-v1-355m": ModelType.IFT,
|
| 26 |
"aisquared/dlite-v1-1_5b": ModelType.IFT,
|
|
|
|
| 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,
|
| 27 |
"aisquared/dlite-v1-355m": ModelType.IFT,
|
| 28 |
"aisquared/dlite-v1-1_5b": ModelType.IFT,
|