search

jxmorris12 / vec2text

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

fix: dtype if using deepspeed #31

Closed zanussbaum closed 6 months ago

zanussbaum commented 6 months ago

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