{ | |
"architectures": [ | |
"TestModel" | |
], | |
"auto_map": { | |
"AutoConfig": "configuration_test.TestConfig", | |
"AutoModel": "modeling_test.TestModel" | |
}, | |
"input_dim": 10, | |
"model_type": "my_test_model", | |
"output_dim": 5, | |
"torch_dtype": "float32", | |
"transformers_version": "4.37.2" | |
} | |