from transformers import AutoModel model = AutoModel.from_pretrained( "RGBD-SOD/custom-resnet50d", trust_remote_code=True ) print(model)