Spaces:
Running
Running
Update README.md and utils.py
Browse files- src/display/utils.py +2 -2
src/display/utils.py
CHANGED
|
@@ -67,7 +67,7 @@ class ModelType(Enum):
|
|
| 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=" "):
|
| 73 |
return f"{self.value.symbol}{separator}{self.value.name}"
|
|
@@ -82,7 +82,6 @@ class ModelType(Enum):
|
|
| 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 |
|
|
@@ -103,6 +102,7 @@ class Precision(Enum):
|
|
| 103 |
return Precision.float16
|
| 104 |
if precision in ["torch.bfloat16", "bfloat16"]:
|
| 105 |
return Precision.bfloat16
|
|
|
|
| 106 |
return Precision.Unknown
|
| 107 |
|
| 108 |
# Column selection
|
|
|
|
| 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=" "):
|
| 73 |
return f"{self.value.symbol}{separator}{self.value.name}"
|
|
|
|
| 82 |
return ModelType.RL
|
| 83 |
if "instruction-tuned" in type or "⭕" in type:
|
| 84 |
return ModelType.IFT
|
|
|
|
| 85 |
if "locked" in type or "🔒" in type:
|
| 86 |
return ModelType.LOCK
|
| 87 |
|
|
|
|
| 102 |
return Precision.float16
|
| 103 |
if precision in ["torch.bfloat16", "bfloat16"]:
|
| 104 |
return Precision.bfloat16
|
| 105 |
+
|
| 106 |
return Precision.Unknown
|
| 107 |
|
| 108 |
# Column selection
|