Skip to content

Add a progress bar for large model loading#20713

Merged
sgugger merged 1 commit intomainfrom
large_model_progress
Dec 12, 2022
Merged

Add a progress bar for large model loading#20713
sgugger merged 1 commit intomainfrom
large_model_progress

Conversation

@sgugger
Copy link
Collaborator

@sgugger sgugger commented Dec 9, 2022

What does this PR do?

As requested in #20669, this PR adds a progress bar when loading large models. The progress bar can be removed with transformers.utils.disable_progress_bar().

Fixes #20669

@sgugger sgugger requested a review from LysandreJik December 9, 2022 17:00
@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Dec 9, 2022

The documentation is not available anymore as the PR was closed or merged.

Copy link
Member

@LysandreJik LysandreJik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Clean!

@sgugger sgugger merged commit 29ff871 into main Dec 12, 2022
@sgugger sgugger deleted the large_model_progress branch December 12, 2022 18:12
mpierrau pushed a commit to mpierrau/transformers that referenced this pull request Dec 15, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Progress Bar for large model loading

3 participants