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 giventransformers_cls
.- Parameters
input_path (
str
) – Path to the pretrained transformers model foldertransformers_cls (
Type
[PreTrainedModel
]) – The class type of the pretrained model you need to load.
- Return type
PreTrainedModel
- Returns