Unverified Commit 566bdf4c authored by bonlime's avatar bonlime Committed by GitHub
Browse files

Allow disabling `from_pretrained` tqdm (#5007)

parent 0eb715d7
...@@ -1079,7 +1079,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): ...@@ -1079,7 +1079,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
from diffusers import pipelines from diffusers import pipelines
# 6. Load each module in the pipeline # 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 # 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 class_name = class_name[4:] if class_name.startswith("Flax") else class_name
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment