Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix argument passthrough for sweep #266

Open
wants to merge 11 commits into
base: main
Choose a base branch
from
Prev Previous commit
Next Next commit
remove unnecessary comments
  • Loading branch information
derpyplops committed Jul 12, 2023
commit 9988cc5cc9704ffc453519c9d7b239f010cd5f01
1 change: 0 additions & 1 deletion elk/training/sweep.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,6 @@ def execute(self):
continue

assert elicit.out_dir is not None
# TODO we should fix this so that this isn't needed

eval = elicit.make_eval(model, eval_dataset)
eval.execute(highlight_color="green")
Expand Down
3 changes: 1 addition & 2 deletions elk/training/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,8 +63,7 @@ def make_eval(self, model, eval_dataset):
skip_supervised=self.supervised == "none",
prompt_indices=self.prompt_indices,
concatenated_layer_offset=self.concatenated_layer_offset,
# datasets=run.datasets isn't needed because it's
# immediately overwritten
# datasets isn't needed because it's immediately overwritten
debug=self.debug,
disable_cache=self.disable_cache,
)
Expand Down