Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update src/display/utils.py
Browse files- src/display/utils.py +2 -2
src/display/utils.py
CHANGED
@@ -108,8 +108,8 @@ class AddSpecialTokens(Enum):
|
|
108 |
Unknown = ModelDetails("?")
|
109 |
|
110 |
class NumFewShots(Enum):
|
111 |
-
shots_0 = ModelDetails("0
|
112 |
-
shots_4 = ModelDetails("4
|
113 |
Unknown = ModelDetails("?")
|
114 |
|
115 |
def from_str(shots):
|
|
|
108 |
Unknown = ModelDetails("?")
|
109 |
|
110 |
class NumFewShots(Enum):
|
111 |
+
shots_0 = ModelDetails("0")
|
112 |
+
shots_4 = ModelDetails("4")
|
113 |
Unknown = ModelDetails("?")
|
114 |
|
115 |
def from_str(shots):
|