diff --git a/llm_benchmark/run_benchmark.py b/llm_benchmark/run_benchmark.py index 7f49e70..6e95d06 100644 --- a/llm_benchmark/run_benchmark.py +++ b/llm_benchmark/run_benchmark.py @@ -116,7 +116,8 @@ def run_benchmark(models_file_path, benchmark_file_path, type, ollamabin: str = print("-"*40) file1.write("\n"+"-"*40) file1.close() - + print (f"Stopping model {model_name}") + subprocess.run([ollamabin, 'stop', model_name], capture_output=False, text=True, check=False, encoding='utf-8') return ans if __name__ == "__main__": @@ -126,4 +127,6 @@ def run_benchmark(models_file_path, benchmark_file_path, type, ollamabin: str = run_benchmark(args.models, args.benchmark, args.type, args.ollamabin) print('-'*40) - \ No newline at end of file + + +