transformers
Adding `mps` as device for `Pipeline` class
#30080
Merged

Adding `mps` as device for `Pipeline` class #30080

amyeroberts merged 4 commits into huggingface:main from fnhirwa:pipeline
fnhirwa
fnhirwa adding env variable for mps and is_torch_mps_available for Pipeline
40ca8423
fnhirwa fix linting errors
a7bcbd4a
amyeroberts
amyeroberts commented on 2024-04-08
fnhirwa Remove environment overide
6ae2f265
fnhirwa fnhirwa requested a review from amyeroberts amyeroberts 1 year ago
amyeroberts
amyeroberts commented on 2024-04-08
amyeroberts
amyeroberts approved these changes on 2024-04-08
fnhirwa Merge remote-tracking branch 'upstream/main' into pipeline
fa2b77f4
HuggingFaceDocBuilderDev
amyeroberts amyeroberts merged 8c00b53e into main 1 year ago
fnhirwa fnhirwa deleted the pipeline branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone