How to use Giatti/test-model with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Giatti/test-model", device_map="auto")
ae3876f
1
2
3
4
5
{ "hidden_size": 32, "num_channels": 3, "num_classes": 10 }
