Fix import of `torchao` now that `torchao-nightly` does not exist

[joecummings]

Remove torchao_nightly package by atalman · Pull Request #661 · pytorch/ao

[jetstudio-io]

I got same problem of torchao version while installing the torchtune version 0.2.1

ERROR: Could not find a version that satisfies the requirement torchao==0.3.1 (from torchtune) (from versions: 0.0.1, 0.0.3, 0.1, 0.4.0)
ERROR: No matching distribution found for torchao==0.3.1

[RdoubleA]

@jetstudio-io I was not able to reproduce your error, as torchao==0.3.1 is available in PyPI. What is the pip install command you are doing?