Update README.md

README.md

@@ -156,11 +156,11 @@ pip install elastic_models[blackwell]\
  --index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple\
  --extra-index-url https://pypi.nvidia.com\
  --extra-index-url https://pypi.org/simple
+pip install -U --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128
+pip install -U --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cu128
 
 pip install flash_attn==2.7.3 --no-build-isolation
 pip uninstall apex
-pip install -U --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128
-pip install -U --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cu128
 pip install tensorrt==10.11.0.33 opencv-python==4.11.0.86 imageio-ffmpeg==0.6.0
 ```