PPE: lazy calflops import so src loads without accelerate on Space

D-FINE/src/misc/profiler_utils.py

@@ -5,13 +5,17 @@ Copyright (c) 2024 The D-FINE Authors. All Rights Reserved.
 import copy
 from typing import Tuple
 
-from calflops import calculate_flops
+# calflops is a training-time FLOPs profiler that pulls in `accelerate`. It is not
+# needed for inference, and the deployment image removes `accelerate`, so import it
+# lazily inside stats() to keep the package importable for inference-only use.
 
 
 def stats(
     cfg,
     input_shape: Tuple = (1, 3, 640, 640),
 ) -> Tuple[int, dict]:
+    from calflops import calculate_flops
+
     base_size = cfg.train_dataloader.collate_fn.base_size
     input_shape = (1, 3, base_size, base_size)