Skip to content

[Fix] Remove calls to unsupported functions in TP - #46103

Merged
ArthurZucker merged 4 commits into
mainfrom
fix-tp-unsupported
May 20, 2026
Merged

[Fix] Remove calls to unsupported functions in TP#46103
ArthurZucker merged 4 commits into
mainfrom
fix-tp-unsupported

Conversation

@remi-or

@remi-or remi-or commented May 20, 2026

Copy link
Copy Markdown
Collaborator

In #45028 some code was introduced that is not compatible with torch < 2.9.1 . Since we support up to torch 2.4 this PR fixes two instances of this and also adds a TODO for check when instantiating a distributed model.

@remi-or
remi-or requested review from 3outeille and ArthurZucker May 20, 2026 07:13
@remi-or remi-or self-assigned this May 20, 2026
@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.

@ArthurZucker
ArthurZucker merged commit 4dbb358 into main May 20, 2026
93 of 96 checks passed
@ArthurZucker
ArthurZucker deleted the fix-tp-unsupported branch May 20, 2026 09:15
yuchenxie4645 pushed a commit to yuchenxie4645/transformers that referenced this pull request May 28, 2026
* Alias without _ to be compatible with torch < 2.10

* Remove the _unflatten call which is not compatible w/ torch < 2.9.1

* Add a TODO and remove an assert
kashif pushed a commit to kashif/transformers that referenced this pull request Jun 1, 2026
* Alias without _ to be compatible with torch < 2.10

* Remove the _unflatten call which is not compatible w/ torch < 2.9.1

* Add a TODO and remove an assert
khushali9 pushed a commit to khushali9/transformers that referenced this pull request Jun 8, 2026
* Alias without _ to be compatible with torch < 2.10

* Remove the _unflatten call which is not compatible w/ torch < 2.9.1

* Add a TODO and remove an assert
stevhliu pushed a commit to stevhliu/transformers that referenced this pull request Jul 30, 2026
* Alias without _ to be compatible with torch < 2.10

* Remove the _unflatten call which is not compatible w/ torch < 2.9.1

* Add a TODO and remove an assert
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