transformers
mps: add `isin_mps_friendly`, a wrapper function for `torch.isin`
#33099
Merged
Overview
Commits
4
Changes
View On GitHub

mps: add `isin_mps_friendly`, a wrapper function for `torch.isin` #33099

gante merged 4 commits into huggingface:main from gante:isin_mps
gante
gante add wrapper fn
5e2a2e8b
HuggingFaceDocBuilderDev
gante torch.isin has an issue when using named arguments
13f8949f
gante gante requested a review from sanchit-gandhi sanchit-gandhi 1 year ago
gante gante requested a review from amyeroberts amyeroberts 1 year ago
gante add test
06877b75
ArthurZucker
ArthurZucker approved these changes on 2024-08-26
gante gated import
eab0c847
gante gante merged 72d4a3f9 into main 1 year ago
gante gante deleted the isin_mps branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum