search

jxmorris12 / vec2text

utilities for decoding deep representations (like sentence embeddings) back to text
Other
686 stars 76 forks source link

Add mps support #3

Closed NMontanaBrown closed 11 months ago

NMontanaBrown commented 11 months ago

Issue Description

MPS device support could be added by checking for "mps" availability prior to assigning to "cpu" by changing conditional statement here:

https://github.com/jxmorris12/vec2text/blob/b121c689402b16728d4aa71362c2eac4374c1c26/vec2text/models/model_utils.py#L36

jxmorris12 commented 11 months ago

LGTM! Thanks! Now people should be able to run this on their Mac, I think!