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