Update README.md

README.md

@@ -25,7 +25,7 @@ This repository hosts the **Phi4-mini-instruct** model quantized with [torchao](
 Install vllm nightly and torchao nightly to get some recent changes:
 ```
 pip install vllm --pre --extra-index-url https://wheels.vllm.ai/nightly
-pip install torchao
+pip install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cu126
 ```
 
 ## Code Example
@@ -78,7 +78,7 @@ Install the required packages:
 ```Shell
 pip install git+https://github.com/huggingface/transformers@main
 pip install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cu126
-pip install torch
+pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu126
 pip install accelerate
 ```
 
@@ -129,7 +129,7 @@ Install the required packages:
 ```Shell
 pip install git+https://github.com/huggingface/transformers@main
 pip install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cu126
-pip install torch
+pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu126
 pip install accelerate
 ```