Update README.md

README.md

@@ -25,7 +25,7 @@ pipeline_tag: text-generation
 First need to install the required packages:
 
 ```
-pip install git+https://github.com/huggingface/transformers
+pip install git+https://github.com/huggingface/transformers@main
 pip install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cu126
 ```