How to use theadityamittal/resnet152 with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("theadityamittal/resnet152", device_map="auto")
