Added tqdm for export and compile

Signed-off-by: Amit Raj <amitraj@qti.qualcommm.com>

Author: Amit Raj

QEfficient/diffusers/pipelines/flux/pipeline_flux.py

@@ -14,6 +14,7 @@
 from diffusers import FluxPipeline
 from diffusers.image_processor import VaeImageProcessor
 from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion import retrieve_timesteps
+from tqdm import tqdm
 
 from QEfficient.diffusers.pipelines.pipeline_module import (
     QEffFluxTransformerModel,
@@ -161,7 +162,7 @@ def export(self, export_dir: Optional[str] = None) -> str:
         Returns:
             str: Path to the export directory
         """
-        for module_name, module_obj in self.modules.items():
+        for module_name, module_obj in tqdm(self.modules.items(), desc="Exporting modules", unit="module"):
             # Get ONNX export configuration for this module
             example_inputs, dynamic_axes, output_names = module_obj.get_onnx_config()
 
@@ -219,7 +220,7 @@ def compile(self, compile_config: Optional[str] = None) -> None:
             config_manager(self, config_source=compile_config)
 
         # Compile each module with its specific configuration
-        for module_name, module_obj in self.modules.items():
+        for module_name, module_obj in tqdm(self.modules.items(), desc="Compiling modules", unit="module"):
             module_config = self.custom_config["modules"]
             specializations = module_config[module_name]["specializations"]
             compile_kwargs = module_config[module_name]["compilation"]