config.json

{
 "architectures": [
 "CustomModel"
 ],
 "auto_map": {
 "AutoConfig": "__main__.CustomConfig",
 "AutoModel": "__main__.CustomModel"
 },
 "model_type": "test-zeroshot",
 "torch_dtype": "float32",
 "transformers_version": "4.23.1"
}