Spaces:
Running
Running
Commit
Β·
e1cdc4b
1
Parent(s):
671e1a6
[FIX] Preference tuned model symbol
Browse files- src/display/utils.py +2 -2
src/display/utils.py
CHANGED
@@ -95,8 +95,8 @@ class ModelType(Enum):
|
|
95 |
# return ModelType.FT
|
96 |
if "pretrained" in type or "π’" in type:
|
97 |
return ModelType.PT
|
98 |
-
|
99 |
-
|
100 |
if "instruction-tuned" in type or "β" in type:
|
101 |
return ModelType.IFT
|
102 |
# if "domain-specific" in type or "β" in type:
|
|
|
95 |
# return ModelType.FT
|
96 |
if "pretrained" in type or "π’" in type:
|
97 |
return ModelType.PT
|
98 |
+
if "preference-tuned" in type or "π¦" in type:
|
99 |
+
return ModelType.RL
|
100 |
if "instruction-tuned" in type or "β" in type:
|
101 |
return ModelType.IFT
|
102 |
# if "domain-specific" in type or "β" in type:
|