pminervini commited on
Commit
43578e7
1 Parent(s): dc1ba50
scripts/create_request_file.py CHANGED
@@ -40,7 +40,7 @@ def main():
40
 
41
  model_name = click.prompt("Enter model name")
42
  revision = click.prompt("Enter revision", default="main")
43
- precision = click.prompt("Enter precision", default="float16", type=click.Choice(precisions))
44
  model_type = click.prompt("Enter model type", type=click.Choice(model_types))
45
  weight_type = click.prompt("Enter weight type", default="Original", type=click.Choice(weight_types))
46
  base_model = click.prompt("Enter base model", default="")
 
40
 
41
  model_name = click.prompt("Enter model name")
42
  revision = click.prompt("Enter revision", default="main")
43
+ precision = click.prompt("Enter precision", default="float32", type=click.Choice(precisions))
44
  model_type = click.prompt("Enter model type", type=click.Choice(model_types))
45
  weight_type = click.prompt("Enter weight type", default="Original", type=click.Choice(weight_types))
46
  base_model = click.prompt("Enter base model", default="")
src/display/utils.py CHANGED
@@ -14,6 +14,7 @@ class Task:
14
  metric: str
15
  col_name: str
16
 
 
17
  class Tasks(Enum):
18
  # arc = Task("arc:challenge", "acc_norm", "ARC")
19
  # hellaswag = Task("hellaswag", "acc_norm", "HellaSwag")
 
14
  metric: str
15
  col_name: str
16
 
17
+
18
  class Tasks(Enum):
19
  # arc = Task("arc:challenge", "acc_norm", "ARC")
20
  # hellaswag = Task("hellaswag", "acc_norm", "HellaSwag")