accelerate
Fix infer_auto_device_map when tied weights share the same prefix name
#2324

Merged

Fix infer_auto_device_map when tied weights share the same prefix name #2324

fxmarty merged 2 commits into main from fix-tied-param-detection

fix auto device map with tied weights sharing a prefix name

7916a494

precise comment

5e780993

fxmarty requested a review from

SunMarc 2 years ago

fxmarty requested a review from

muellerzr 2 years ago

muellerzr approved these changes on 2024-01-10

fxmarty merged e3e9b875 into main 2 years ago

fxmarty deleted the fix-tied-param-detection branch 2 years ago

SunMarc commented on 2024-01-10

Reviewers

muellerzr

SunMarc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone