search

jxmorris12 / vec2text

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

fix: dtype if using deepspeed #31

Closed zanussbaum closed 9 months ago

zanussbaum commented 9 months ago

deepspeed sets the weights of the model in mixed precision, need to change precision of embeddings