model_head/config.json

{"in_features": 768, "out_features": 2, "temperature": 1.0, "bias": true, "device": "cuda"}