nlpstack.integrations.transformers.cache module#

nlpstack.integrations.transformers.cache.clear()[source]#
Return type:

None

nlpstack.integrations.transformers.cache.get_pretrained_model(pretrained_model_name_or_path, auto_cls=None)[source]#
Parameters:
  • pretrained_model_name_or_path (str | PathLike) –

  • auto_cls (Type | None) –

Return type:

transformers.PretrainedModel

nlpstack.integrations.transformers.cache.get_pretrained_tokenizer(pretrained_model_name_or_path)[source]#
Return type:

PreTrainedTokenizer

Parameters:

pretrained_model_name_or_path (str | PathLike) –