The Command returned non-zero code: 137

Hi, I am using sm-docker to build my custom environment image in aws sagemaker studio. However while solving environment it gets Killed with error code 137. I tried to do research and found out, that can be because of memory limit and we can specify memory in docker command. I am not sure how to do that with sm-docker command. Is there anything already there like passing an argument for that which I am missing? Thanks.