Skip to content

Conversation

@patrickvonplaten
Copy link
Contributor

@patrickvonplaten patrickvonplaten commented Nov 3, 2022

This PR corrects the naming to the same we have in Transformers and additionally:

  • Makes sure if one intentionally sets low_cpu_mem_usage=True if PyTorch < 1.9 is installed => we raise an Error
  • Makes sure we raise an error if low_cpu_mem_usage=False but device_map=True

@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Nov 3, 2022

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

@patrickvonplaten patrickvonplaten changed the title correct naming [Low cpu memory] Correct naming and improve default usage Nov 3, 2022
Copy link
Contributor

@patil-suraj patil-suraj left a comment

Choose a reason for hiding this comment

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

Looks good, thanks for fixing this!

@patrickvonplaten patrickvonplaten merged commit 42bb459 into main Nov 3, 2022
@patrickvonplaten patrickvonplaten deleted the fast_load_to_low_cpu_mem_usage branch November 3, 2022 17:11
yoonseokjin pushed a commit to yoonseokjin/diffusers that referenced this pull request Dec 25, 2023
…e#1122)

* correct naming

* finish

* Apply suggestions from code review

* Apply suggestions from code review

Co-authored-by: Suraj Patil <[email protected]>

Co-authored-by: Suraj Patil <[email protected]>
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.

4 participants