diff --git a/magemaker/runner.py b/magemaker/runner.py index 6de523e..0b89a75 100644 --- a/magemaker/runner.py +++ b/magemaker/runner.py @@ -166,11 +166,14 @@ def runner(): help="Deploy a Hugging Face Model.", type=str ) + parser.add_argument( - "--instance", - help="EC2 instance type to deploy to.", + "--deployment_instance", + help="EC2 instance type to deploy the model to.", type=str ) + + parser.add_argument( "--deploy", help="path to YAML deployment configuration file",