Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update src/display/utils.py
Browse files- src/display/utils.py +4 -0
src/display/utils.py
CHANGED
@@ -101,6 +101,10 @@ class Precision(Enum):
|
|
101 |
return Precision.bfloat16
|
102 |
return Precision.Unknown
|
103 |
|
|
|
|
|
|
|
|
|
104 |
# Column selection
|
105 |
COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
|
106 |
|
|
|
101 |
return Precision.bfloat16
|
102 |
return Precision.Unknown
|
103 |
|
104 |
+
class NumFewShotsType(Enum):
|
105 |
+
shot0 = ModelDetails("0-shots")
|
106 |
+
shot4 = ModelDetails("4-shots")
|
107 |
+
|
108 |
# Column selection
|
109 |
COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
|
110 |
|