orkgnlp.common.util.io.load_transformers_pretrained

load_transformers_pretrained(input_path, transformers_cls)[source]

Loads a transformers model weights given of the given input_path to the given transformers_cls.

Parameters:

  • input_path (str) – Path to the pretrained transformers model folder

  • transformers_cls (Type[PreTrainedModel]) – The class type of the pretrained model you need to load.

Return type:

PreTrainedModel

Returns: