1from transformers import AutoModel, AutoConfig
2
3# Load the model
4model = AutoModel.from_pretrained("liangshuxin/domainbed-vrex-coloredmnist-test1")
5
6# Get model config
7config = AutoConfig.from_pretrained("liangshuxin/domainbed-vrex-coloredmnist-test1")