Adapted from togethercomputer/RedPajama-INCITE-Base-3B-v1 to run in AWS Sagemaker