diff --git a/Makefile b/Makefile index 0cc3b01..9b81d10 100644 --- a/Makefile +++ b/Makefile @@ -5,4 +5,5 @@ venv/touchfile: requirements.txt . venv/bin/activate; pip install -Ur requirements.txt touch venv/touchfile clean: - rm -rf venv \ No newline at end of file + rm -rf venv + \ No newline at end of file diff --git a/magemaker/runner.py b/magemaker/runner.py index 6de523e..f565319 100644 --- a/magemaker/runner.py +++ b/magemaker/runner.py @@ -44,11 +44,19 @@ def runner(): help="Deploy a Hugging Face Model.", type=str ) + parser.add_argument( "--instance", help="EC2 instance type to deploy to.", type=str ) + + parser.add_argument( + "--cpu", + help="specify cpu type", + type=str + ) + parser.add_argument( "--deploy", help="path to YAML deployment configuration file", @@ -161,6 +169,8 @@ def runner(): help="Show magemaker version and exit" ) + + parser.add_argument( "--hf", help="Deploy a Hugging Face Model.",