How to Create Model in SageMaker Console from .tar.gz

Amazon SageMaker
7

Hi @dbb :wave:

In this thread, @philschmid proposed a workaround by adding an environment variable (as below) that worked pretty well on my side.

client.create_model(
    ModelName = self.model_name,
    Containers = [
        {
            'Image': image_uri,
            'Mode': 'SingleModel',
            'ModelDataUrl': f'{model_data_path}/{self.model_file}',
            'Environment': {"MMS_DEFAULT_WORKERS_PER_MODEL": '1'}
        }
    ],
    ExecutionRoleArn = role
)
1 Like