Chronos-2: Only pin_memory when device type is cuda#431

abdulfatir merged 2 commits intoamazon-science:mainamazon-science/chronos-forecasting:mainfrom

abdulfatir:fix-pin-memoryabdulfatir/chronos-forecasting:fix-pin-memoryCopy head branch name to clipboard