You should use CustomClass.from_pretrained. PreTrainedModel.from_pretrained won’t work directly.
CustomClass.from_pretrained
PreTrainedModel.from_pretrained