spacy-curated-transformers icon indicating copy to clipboard operation
spacy-curated-transformers copied to clipboard

Solving the problem of requiring all two tensors to be on different d…

Open 323135jx opened this issue 1 year ago • 3 comments

…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.

323135jx avatar Nov 30 '23 07:11 323135jx

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

323135jx avatar Nov 30 '23 07:11 323135jx

This may be better

323135jx avatar Nov 30 '23 10:11 323135jx

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.

adrianeboyd avatar Dec 11 '23 06:12 adrianeboyd