Spaces:
Runtime error
Runtime error
Update README.md and utils.py
Browse files- src/display/utils.py +5 -0
src/display/utils.py
CHANGED
|
@@ -66,6 +66,7 @@ class ModelType(Enum):
|
|
| 66 |
FT = ModelDetails(name="fine-tuned", symbol="🔶")
|
| 67 |
IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
|
| 68 |
RL = ModelDetails(name="RL-tuned", symbol="🟦")
|
|
|
|
| 69 |
Unknown = ModelDetails(name="", symbol="?")
|
| 70 |
|
| 71 |
def to_str(self, separator=" "):
|
|
@@ -81,6 +82,10 @@ class ModelType(Enum):
|
|
| 81 |
return ModelType.RL
|
| 82 |
if "instruction-tuned" in type or "⭕" in type:
|
| 83 |
return ModelType.IFT
|
|
|
|
|
|
|
|
|
|
|
|
|
| 84 |
return ModelType.Unknown
|
| 85 |
|
| 86 |
class WeightType(Enum):
|
|
|
|
| 66 |
FT = ModelDetails(name="fine-tuned", symbol="🔶")
|
| 67 |
IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
|
| 68 |
RL = ModelDetails(name="RL-tuned", symbol="🟦")
|
| 69 |
+
LOCK = ModelDetails(name="locked", symbol="🔒")
|
| 70 |
Unknown = ModelDetails(name="", symbol="?")
|
| 71 |
|
| 72 |
def to_str(self, separator=" "):
|
|
|
|
| 82 |
return ModelType.RL
|
| 83 |
if "instruction-tuned" in type or "⭕" in type:
|
| 84 |
return ModelType.IFT
|
| 85 |
+
|
| 86 |
+
if "locked" in type or "🔒" in type:
|
| 87 |
+
return ModelType.LOCK
|
| 88 |
+
|
| 89 |
return ModelType.Unknown
|
| 90 |
|
| 91 |
class WeightType(Enum):
|