Skip to content

Commit ee0797f

Browse files
committed
rename variable
1 parent 9633021 commit ee0797f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

evals/sampler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def update_results(dataset, generation_args: dict, results: dict):
5353
model_name = args.get("model_url")
5454
load_8_bits = args.get("load_8_bits")
5555
model = Inference(model_name, load_8_bits)
56-
dataset = load_dataset(DATASET)
56+
dataset = load_dataset(DATA)
5757
generation_args = yaml.safe_load(Path("evals/config/generation.yaml").read_text())
5858
default_args = generation_args.pop("defaults")
5959
generation_args = merge_dicts(generation_args, default_args)

0 commit comments

Comments
 (0)