PR #1115 default fast model loading 🔥

default fast model loading 🔥 #1115

patil-suraj merged 20 commits into main from fast-init

make accelerate hard dep

bbdaab49

default fast init

b6d59beb

move params to cpu when device map is None

29146e17

handle device_map=None

df4c8929

handle torch < 1.9

736e3daa

remove device_map="auto"

bc0a02f5

patil-suraj requested a review from

patrickvonplaten 3 years ago

patil-suraj requested a review from

anton-l 3 years ago

patil-suraj requested a review from

pcuenca 3 years ago

patrickvonplaten marked this pull request as ready for review 3 years ago

patrickvonplaten commented on 2022-11-03

anton-l approved these changes on 2022-11-03

patrickvonplaten commented on 2022-11-03

style

3a5c3259

pcuenca approved these changes on 2022-11-03

add accelerate in torch extra

41b44128

remove accelerate from extras["test"]

d670d8e1

raise an error if torch is available but not accelerate

09cb109b

update installation docs

05a8f411

Apply suggestions from code review

3c8b6786

improve defautl loading speed even further, allow disabling fats loading

7ee3f2f5

address review comments

754d3761

adapt the tests

bb93884e

patil-suraj commented on 2022-11-03

pcuenca approved these changes on 2022-11-03

fix test_stable_diffusion_fast_load

f77100af

fix test_read_init

3207ae50

anton-l commented on 2022-11-03

patil-suraj commented on 2022-11-03

temp fix for dummy checks

441eaa7c

Trigger Build

b461ad3b

Apply suggestions from code review

15ec8599

patil-suraj commented on 2022-11-03

patil-suraj merged 74821781 into main 3 years ago

patil-suraj deleted the fast-init branch 3 years ago

Reviewers

anton-l

pcuenca

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

diffusers default fast model loading 🔥 #1115 Merged

default fast model loading 🔥 #1115

diffusers
default fast model loading 🔥
#1115

Merged