Hugging Face Hubにコードからログインする方法

この記事は公開から1年以上経過しています。

Huggin Face HubにCliからではなくコードでログインする方法。

問題

Hugging FaceのTransformersでfrom_pretrained(use_auth_token=’xxx’)のようにuse_auth_tokenを使用すると

FutureWarning: The use_auth_token argument is deprecated and will be removed in v5 of Transformers.

のような警告が発生する。

原因

from_pretrained(use_auth_token=’xxx’)はTransformers v5で廃止予定となっているため。

対応

huggingface_hubのlogin()を使用する。

pip:

huggingface_hub

main.py

from huggingface_hub import login

login(token='Hugging Face Token Id')

但し、use_auth_tokenと同様、ソースコードにハードコーディングすることによるトークンID漏洩事故を起こさないように注意が必要です。

以上です。