aws-samples / amazon-sagemaker-generativeai

Repository for training and deploying Generative AI models, including text-text, text-to-image generation and prompt engineering playground using SageMaker Studio.
MIT No Attribution
130 stars 88 forks source link

Train and deploy gptj with SageMaker's Model Parallel Tensor Parallelism #4

Closed anapt closed 1 year ago

anapt commented 1 year ago

Issue #, if available: N/A

Description of changes: Code and notebooks for Lab 3. Train and Deploy GPT-J with SageMaker Tensor Parallel

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.