AWS Trainium fails number of device validation when using more than 1 accelerator on the instances

AWS Trainium fails number of device validation when using more than 1 accelerator on the instances #1972