MinishLab / model2vec

Distill a Small Static Model from any Sentence Transformer
https://minishlab.github.io/
MIT License
413 stars 18 forks source link

fix: move tensor to cpu #86

Closed stephantul closed 1 month ago

stephantul commented 1 month ago

We didn't move the tensor to cpu before trying to dump it, which causes an error if the model is on another device.

codecov[bot] commented 1 month ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Files with missing lines Coverage Δ
model2vec/model.py 96.74% <ø> (ø)