add flush() add appropriate places

huggingface · Dec 4, 2023 · de739fa · de739fa
1 parent 0f4ae4e
commit de739fa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/benchmarks/base_classes.py b/benchmarks/base_classes.py
@@ -98,12 +98,14 @@ def benchmark(self, args):
         benchmark_info = BenchmarkInfo(time=time, memory=memory)
 
         pipeline_class_name = str(self.pipe.__class__.__name__)
+        flush()
         csv_dict = generate_csv_dict(
             pipeline_cls=pipeline_class_name, ckpt=args.ckpt, args=args, benchmark_info=benchmark_info
         )
         filepath = self.get_result_filepath(args)
         write_to_csv(filepath, csv_dict)
         print(f"Logs written to: {filepath}")
+        flush()
 
 
 class ImageToImageBenchmark(TextToImageBenchmark):