UTMOSのインストール
非公式実装ですが,pytorchで動かしやすそうなので...
https://github.com/fakerybakery/utmos
Fairseqをgithubから持ってくる
uv add git+https://github.com/facebookresearch/fairseq@v0.10.2
utmosをインストール
uv add utmos
Fairseqのライブラリをいじる
-
.venv/lib/pythonxx/fairseq/data/indexed_dataset.pyの101行目と306行目にあるnp.floatの部分をnp.float32に変更
-
.venv/lib/pythonxx/fairseq/checkpoint_utils.pyの225行目に"weights_only=True"を追加
で多分動く