what's happening.

Author: sayakpaul

benchmarks/benchmarking_utils.py

@@ -9,13 +9,9 @@
 import torch.utils.benchmark as benchmark
 
 from diffusers.models.modeling_utils import ModelMixin
-from diffusers.utils import logging
 from diffusers.utils.testing_utils import require_torch_gpu, torch_device
 
 
-logger = logging.get_logger(__name__)
-
-
 def benchmark_fn(f, *args, **kwargs):
     t0 = benchmark.Timer(
         stmt="f(*args, **kwargs)",
@@ -105,7 +101,7 @@ def post_benchmark(self, model):
     @torch.no_grad()
     def run_benchmark(self, scenario: BenchmarkScenario):
         # 0) Basic stats
-        logger.info(f"Running scenario: {scenario.name}.")
+        print(f"Running scenario: {scenario.name}.")
         model = model_init_fn(scenario.model_cls, **scenario.model_init_kwargs)
         num_params = round(calculate_params(model) / 1e6, 2)
         flops = round(calculate_flops(model, input_dict=scenario.get_model_input_dict()) / 1e6, 2)
@@ -125,7 +121,7 @@ def run_benchmark(self, scenario: BenchmarkScenario):
                 compile_kwargs=None,
             )
         except Exception as e:
-            logger.error(f"Benchmark could not be run with the following error\n: {e}")
+            print(f"Benchmark could not be run with the following error\n: {e}")
             return results
 
         # 2) compiled stats (if any)
@@ -140,7 +136,7 @@ def run_benchmark(self, scenario: BenchmarkScenario):
                     compile_kwargs=scenario.compile_kwargs,
                 )
             except Exception as e:
-                logger.error(f"Compilation benchmark could not be run with the following error\n: {e}")
+                print(f"Compilation benchmark could not be run with the following error\n: {e}")
                 if plain is None:
                     return results
 
@@ -170,10 +166,10 @@ def run_bencmarks_and_collate(self, scenarios: Union[BenchmarkScenario, list[Ben
             try:
                 records.append(self.run_benchmark(s))
             except Exception as e:
-                logger.error(f"Running scenario ({s.name}) led to error:\n{e}")
+                print(f"Running scenario ({s.name}) led to error:\n{e}")
         df = pd.DataFrame.from_records([r for r in records if r])
         df.to_csv(filename, index=False)
-        logger.info(f"Results serialized to {filename=}.")
+        print(f"Results serialized to {filename=}.")
 
     def _run_phase(
         self,

benchmarks/run_all.py

@@ -4,17 +4,12 @@
 
 import pandas as pd
 
-from diffusers.utils import logging
-
 
 PATTERN = "benchmarking_*.py"
 FINAL_CSV_FILENAME = "collated_results.csv"
 GITHUB_SHA = os.getenv("GITHUB_SHA", None)
 
 
-logger = logging.get_logger(__name__)
-
-
 class SubprocessCallException(Exception):
     pass
 
@@ -42,12 +37,12 @@ def run_scripts():
 
     for file in python_files:
         if file != "benchmarking_utils.py":
-            logger.info(f"****** Running file: {file} ******")
+            print(f"****** Running file: {file} ******")
             command = f"python {file}"
             try:
                 run_command(command.split())
             except SubprocessCallException as e:
-                logger.error(f"Error running {file}: {e}")
+                print(f"Error running {file}: {e}")
                 continue