Support both dtype and torch_dtype for older transformers versions (#505)

*Issue #, if available:* Fixes #504. 

*Description of changes:*


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

Co-authored-by: Abdul Fatir Ansari <ansarnd@amazon.de>