fix save name

shahules786 · shahules786 · commit c2627098ced1 · 2023-06-27T09:39:51.000+05:30
diff --git a/evals/sampler.py b/evals/sampler.py
@@ -71,5 +71,5 @@ def update_results(dataset, generation_args: dict, results: dict):
         "prompts":[]
     }
     results = update_results(dataset, generation_args, results)
-    print(results)
+    model_name = model_name.split("/")[-1]
     save_json(f"results-{model_name}.json", results)

Original file line number	Diff line number	Diff line change
`@@ -71,5 +71,5 @@ def update_results(dataset, generation_args: dict, results: dict):`
`71`	`71`	`"prompts":[]`
`72`	`72`	`}`
`73`	`73`	`results = update_results(dataset, generation_args, results)`
`74`		`- print(results)`
	`74`	`+ model_name = model_name.split("/")[-1]`
`75`	`75`	`save_json(f"results-{model_name}.json", results)`