Fixing TorchAO when Downgrading PyTorch for AITune

How I fixed TorchAO compatibility when downgrading PyTorch for AITune