Update README.md

README.md

@@ -23,7 +23,11 @@ pipeline_tag: text-generation
 Install vllm 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 torch --index-url https://download.pytorch.org/whl/nightly/cu126
+```
+To use fbgemm kernels:
+```Shell
+pip install --pre fbgemm-gpu-genai --index-url https://download.pytorch.org/whl/nightly/cu126
 ```
 
 ## Code Example
@@ -72,11 +76,15 @@ vllm serve pytorch/Phi-4-mini-instruct-FP8 --tokenizer microsoft/Phi-4-mini-inst
 Install the required packages:
 ```Shell
 pip install git+https://github.com/huggingface/transformers@main
-pip install torchao
-pip install torch
+pip install --pre torchao torch --index-url https://download.pytorch.org/whl/nightly/cu126
 pip install accelerate
 ```
 
+To use fbgemm kernels:
+```Shell
+pip install --pre fbgemm-gpu-genai --index-url https://download.pytorch.org/whl/nightly/cu126
+```
+
 Example:
 ```Py
 import torch