見出し画像

Huggingface Transformers 入門 (21) - モデルのキャッシュパス

npaka

「Huggingface Transformers」のモデルのキャッシュパスについてまとめました。

・Huggingface Transformers 4.4.2

前回

1. モデルのデフォルトのキャッシュパス

「Huggingface Transformers」のモデルは、初回利用時にダウンロードおよびキャッシュされます。デフォルトのキャッシュパスは環境ごとに異なります。

以下のコードで、モデルのデフォルトのキャッシュパスを確認できます。

# モデルのキャッシュパスの確認
from transformers import file_utils
print(file_utils.default_cache_path)

該当フォルダを削除することで、モデルのキャッシュを削除できます。

2. モデルのキャッシュパスの変更

以下のコードで、モデルのキャッシュパスを変更できます。

# モデルのキャッシュパスの変更
import os
os.environ['TRANSFORMERS_CACHE'] = 'cache/'

【おまけ】 Sentence Transformersのキャッシュパスの変更

以下のコードで、「Sentence Transformers」のキャッシュパスを変更できます。

import os
os.environ['SENTENCE_TRANSFORMERS_HOME'] = 'cache/'

次回



この記事が気に入ったらサポートをしてみませんか？