HuggingFace Transformers is remapping `torch_dtype` to `dtype`. We just need to refactor our codebase to comply with this change.