Skip to content

[core] use swap_tensors in group offloading where possible#13751

Open
sayakpaul wants to merge 3 commits into
mainfrom
group-offloading-swap-tensors
Open

[core] use swap_tensors in group offloading where possible#13751
sayakpaul wants to merge 3 commits into
mainfrom
group-offloading-swap-tensors

Conversation

@sayakpaul
Copy link
Copy Markdown
Member

@sayakpaul sayakpaul commented May 14, 2026

What does this PR do?

See #13276 (comment). .data is a private attribute and swap_tensors() should be used instead of it.

I have run the full group offloading tests across the board and they pass.

@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

hooks size/M PR with diff < 200 LOC

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants