Update README.md

README.md

@@ -23,7 +23,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
 ```