accelerate
fix(modeling): include named_buffers in module split expansion for infer_auto_device_map
#4020
Closed
Closed
fix(modeling): include named_buffers in module split expansion for infer_auto_device_map #4020
Anai-Guo
wants to merge
1 commit into
huggingface:main
from
Anai-Guo:fix-infer-auto-device-map-missing-buffers
fix(modeling): include named_buffers in module split expansion for in…
96d13a9f
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub