Update README.md

README.md

@@ -27,24 +27,23 @@ metrics:
 
 ## Setup
 
-You can run the smashed model on a A100 GPU by:
-1. Installing and importing the `pruna-engine` (version 0.2.7) 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 detailed on the package.
-2. Downloading the model files at `model_path`. This can be done using huggingface with this repository name or with manual downloading.
-3. Loading the model
+You can run the smashed model by:
+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
+    huggingface-cli download PrunaAI/segmind-SSD-1B-smashed --local-dir segmind-SSD-1B-smashed --local-dir-use-symlinks False
+    ```
+   Alternatively, you can download them manually.
+3. Loading the model.
 4. Running the model.
-
-The model is smashed for 1024x1024 image size. Contact us if you want a version with dynamic size support.
-
 You can achieve this by running the following code:
 
 ```python
-from transformers.utils.hub import cached_file
 from pruna_engine.PrunaModel import PrunaModel  # Step (1): install and import `pruna-engine` package.
-
-...
-model_path = cached_file("PrunaAI/segmind-SSD-1B-smashed", "model")  # Step (2): download the model files at `model_path`.
+model_path = "segmind-SSD-1B-smashed/model" # Step (2): specify the downloaded model path.
 smashed_model = PrunaModel.load_model(model_path)  # Step (3): load the model.
-y = smashed_model(prompt="a silly prune with a face in high definition", image_height=1024, image_width=1024)[0]  # Step (4): run the model.
+y = smashed_model(prompt="an astronaut riding a horse on mars", image_height=1024, image_width=1024)[0]  # Step (4): run the model.
 ```
 
 ## Configurations