spacy-curated-transformers
spacy-curated-transformers copied to clipboard
Solving the problem of requiring all two tensors to be on different d…
…evices in the Flask environment
Description
Types of change
Checklist
- [x] I confirm that I have the right to submit this contribution under the project's MIT license.
- [x] I ran the tests, and all new and existing tests passed.
- [x] My changes don't require a change to the documentation, or if they do, I've added all required information.
I find this problem in zh_core_web_trf 3.2.0 upgradation to 3.7.2,this code references spacy_transformers line 214-215
This may be better
Thanks for the PR! We think we understand where the underlying problem is, but we need to look into the details a bit more to make sure that we have a solution that's general enough to also support other devices like MPS.