EleutherAI / elk

Keeping language models honest by directly eliciting knowledge encoded in their activations.
MIT License
175 stars 32 forks source link

Fix sweep ds config bug #279

Closed reaganjlee closed 11 months ago

reaganjlee commented 12 months ago

Previously, running sweep with datasets with config names, such as super_glue:boolq, would error as Sweep would check for datasets existing without partitioning for configs as we do for elicit, which this now adds support for

Reproducing ex elk sweep --models bigscience/bloomz-7b1-mt --datasets super_glue:boolq