transformers
c31473ed
- Remove float64 cast for OwlVit and OwlV2 to support MPS device (#31071)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Remove float64 cast for OwlVit and OwlV2 to support MPS device (#31071) Remove float64
References
#31071 - Remove float64 cast for OwlVit and OwlV2 to support MPS device
Author
qubvel
Parents
936ab7ba
Loading