How to change huggingface transformers default cache directory

by

You can specify the cache directory everytime you load a model with .from_pretrained by the setting the parameter cache_dir. You can define a default location by exporting an environment variable TRANSFORMERS_CACHE everytime before you use (i.e. before importing it!) the library).

Example for python:

import os
os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'

Example for bash:

export TRANSFORMERS_CACHE=/blabla/cache/

More Related Contents:

  1. BERT sentence embeddings from transformers

Leave a Comment