transformers
Remove float64 cast for OwlVit and OwlV2 to support MPS device
#31071
Merged

Remove float64 cast for OwlVit and OwlV2 to support MPS device #31071

qubvel merged 1 commit into huggingface:main from qubvel:fix-mps-float64
qubvel
qubvel Remove float64
a3b3e748
HuggingFaceDocBuilderDev
amyeroberts
amyeroberts approved these changes on 2024-05-28
qubvel qubvel merged c31473ed into main 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone