Update README.md

README.md

@@ -29,19 +29,23 @@ Above results for 768x768 image size with 4-step inference on A100. Dynamic imag
 ## 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 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
-4. Running the model. 
+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.
+2. Downloading the model files. This can be done using the Hugging Face CLI with the following commands:
+ ```bash
+ mkdir SimianLuo-LCM_Dreamshaper_v7-smashed
+ huggingface-cli download SimianLuo-LCM_Dreamshaper_v7-smashed --local-dir PrunaAI/SimianLuo-LCM_Dreamshaper_v7-smashed --local-dir-use-symlinks False
+ ```
+ Alternatively, you can download them manually.
+3. Loading the model.
+4. Running the model.
 
 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/SimianLuo-LCM_Dreamshaper_v7-smashed", "model") # Step (2): download the model files at `model_path`.
-smashed_model = PrunaModel.load_model(model_path) # Step (3): load the model.
+model_path = "PrunaAI/SimianLuo-LCM_Dreamshaper_v7-smashed" # Step (2): specify the downloaded model path.
+smashed_model = PrunaModel.load_model('SimianLuo-LCM_Dreamshaper_v7-smashed/model') # Step (3): load the model.
 smashed_model("Self-portrait oil painting, a beautiful cyborg with golden hair, 8k", num_inference_steps=4)[0] # Step (4): run the model.
 ```