transformers

d7633a4e - Add basic support for FP16 in SageMaker model parallelism (#11407)

Add basic support for FP16 in SageMaker model parallelism (#11407) * Add FP16 support for SageMaker MP * Add print debugs * Squeeze * Remove debug statements * Add defensive check * Typo