huggingface transformers cache directory 변경(웨이트 저장되는 폴더 변경) 방법

import os

os.environ["HF_HOME"] = "/workspace/"


import torch


from transformers import pipeline

transformers import하기전에 환경변수 HF_HOME을 원하는 위치로 지정하면 된다.

 

export나 bashrc에 기록하는 것도 방법이다.

https://daeun-computer-uneasy.tistory.com/145?category=1080901

[Linux] Huggingface model default 저장 경로 (./.cache) 변경하기 (HF_HOME 지정)