CI fix for torch 2.1 release (#4452)

Commit

2 years ago

CI fix for torch 2.1 release (#4452) * Fix for torch 2.1 release Co-authored-by: Logan Adams <loadams@microsoft.com>