Update README.md

README.md

@@ -28,7 +28,7 @@ metrics:
 ## Setup
 
 You can run the smashed model by:
-1. Installing and importing the `pruna-engine` (version 0.2.9) package. Use `pip install pruna --extra-index-url https://pypi.nvidia.com --extra-index-url https://pypi.ngc.nvidia.com` for installation. See [Pypi](https://pypi.org/project/pruna-engine/) for details on the package.
+1. Installing and importing the `pruna-engine` (version 0.2.9) package. Use `pip install pruna-engine==0.2.9 --extra-index-url https://pypi.nvidia.com --extra-index-url https://pypi.ngc.nvidia.com` for installation. See [Pypi](https://pypi.org/project/pruna-engine/) for details on the package.
 2. Downloading the model files. This can be done using the Hugging Face CLI with the following commands:
     ```bash
     mkdir CompVis-stable-diffusion-v1-4-smashed