KeyError: 'channels'

[Boom5426]

If I just want to use it in other field, what should I do?

import torch
model = torch.hub.load('insitro/ChannelViT', 'cpjump_cellpaint_channelvit_small_p8_with_hcs_supervised', pretrained=True)
model.eval()
images = torch.randn(5, 3, 224, 224)
out = model(images)

KeyError: 'channels'

[Boom5426]

What param shoud I input the extra_tokens?