From 566bdf4c4432caacd5317aa1fdb361333d40dfba Mon Sep 17 00:00:00 2001 From: bonlime Date: Thu, 14 Sep 2023 17:36:19 +0300 Subject: [PATCH] Allow disabling `from_pretrained` tqdm (#5007) --- src/diffusers/pipelines/pipeline_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py index 63e25fdf75..b67e823627 100644 --- a/src/diffusers/pipelines/pipeline_utils.py +++ b/src/diffusers/pipelines/pipeline_utils.py @@ -1079,7 +1079,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): from diffusers import pipelines # 6. Load each module in the pipeline - for name, (library_name, class_name) in tqdm(init_dict.items(), desc="Loading pipeline components..."): + for name, (library_name, class_name) in logging.tqdm(init_dict.items(), desc="Loading pipeline components..."): # 6.1 - now that JAX/Flax is an official framework of the library, we might load from Flax names class_name = class_name[4:] if class_name.startswith("Flax") else class_name