Unverified Commit 69ed600d authored by Poedator's avatar Poedator Committed by GitHub
Browse files

fix for non-str output from evaluator

parent b9a6a3df
...@@ -102,7 +102,7 @@ def simple_evaluate( ...@@ -102,7 +102,7 @@ def simple_evaluate(
# add info about the model and few shot config # add info about the model and few shot config
results["config"] = { results["config"] = {
"model": model, "model": (model if isinstance(model, str) else model.config._name_or_path),
"model_args": model_args, "model_args": model_args,
"num_fewshot": num_fewshot, "num_fewshot": num_fewshot,
"batch_size": batch_size, "batch_size": batch_size,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment