Update README.md

README.md

@@ -36,7 +36,7 @@ On iPhone 15 Pro, the model runs at 14.8 tokens/sec and uses 3379 Mb of memory.
 First need to install the required packages:
 ```Shell
 pip install git+https://github.com/huggingface/transformers@main
-pip install torchao
+pip install git+https://github.com/pytorch/ao.git@main
 ```
 
 ## Untie Embedding Weights