Update README.md

README.md

@@ -36,7 +36,7 @@ On iPhone 15 Pro, the model runs at 17.3 tokens/sec and uses 3206 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 --pre torchao torch --index-url https://download.pytorch.org/whl/nightly/cu126
 ```
 
 ## Untie Embedding Weights