From 9cef07da5a1f28ddfeadd232ede9a815c7945aec Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Sun, 17 Dec 2023 15:32:11 +0530 Subject: [PATCH] [Benchmarks] fix: lcm benchmarking reporting (#6198) * fix: lcm benchmarking reporting * fix generate_csv_dict call. --- benchmarks/base_classes.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/benchmarks/base_classes.py b/benchmarks/base_classes.py index 5d328f62b9..38e7663fbf 100644 --- a/benchmarks/base_classes.py +++ b/benchmarks/base_classes.py @@ -162,6 +162,25 @@ class LCMLoRATextToImageBenchmark(TextToImageBenchmark): guidance_scale=1.0, ) + def benchmark(self, args): + flush() + + print(f"[INFO] {self.pipe.__class__.__name__}: Running benchmark with: {vars(args)}\n") + + time = benchmark_fn(self.run_inference, self.pipe, args) # in seconds. + memory = bytes_to_giga_bytes(torch.cuda.max_memory_allocated()) # in GBs. + 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=self.lora_id, 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): pipeline_class = AutoPipelineForImage2Image