Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
pminervini
commited on
Commit
•
43578e7
1
Parent(s):
dc1ba50
update
Browse files- scripts/create_request_file.py +1 -1
- src/display/utils.py +1 -0
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="
|
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")
|