Closed2023/10/20にクローズ1

SentenceTransformerでログもなくエラーで停止して困った話

結論

PyTorchのバージョンを下げた。

pip install torch==2.0.1

SentenceTransformerをいじっていて以下のようなコードを書いた

from sentence_transformers import SentenceTransformer

# モデルのロード
model = SentenceTransformer("sentence-transformers/stsb-xlm-r-multilingual")

# ベクトル化
vector = model.encode(["test", "test2", "test3"])

ほとんど何も出てないけど、処理がそのまま落ちてしまっていたが
以下のようなログだけが出ていた

Segmentation fault

ログをググってみると・・・

以下のIssueにたどり着き、見てみるとPyTorchのバージョンで出る事があるらしい

以前うまくいっていたPyTorchのバージョンを確認

torch==2.0.1

というわけでtorchのバージョンを指定してインストールしたらうまくいった

pip install torch==2.0.1

このスクラップは2023/10/20にクローズされました