DeepSpeedExamples
DeepSpeedExamples copied to clipboard
Create GPT-Neo on CPU and set the correct device when running on gpus