[console_scripts]
torch-model-state = torch_model_state:cli_main

