t0-0 commited on
Commit
50fff65
·
verified ·
1 Parent(s): 38ca5c3

Update src/display/utils.py

Browse files
Files changed (1) hide show
  1. 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 shots")
112
- shots_4 = ModelDetails("4 shots")
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):