How to use layel2/pytorch-model2 with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("layel2/pytorch-model2", device_map="auto")
6fd4d69
1
2
3
4
5
{ "num_channels": 3, "hidden_size": 32, "num_classes": 10 }
