PrunaAI
/

stabilityai-stable-diffusion-2-1-base-turbo-green-smashed

Model card Files Files and versions Community

sharpenb commited on Jan 26

Commit

8905e7b

•

1 Parent(s): 49048f9

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -32,7 +32,7 @@ You can run the smashed model by:
 2. Downloading the model files. This can be done using the Hugging Face CLI with the following commands:
     ```bash
     mkdir stabilityai-stable-diffusion-2-1-base-smashed
-    huggingface-cli download PrunaAI/stabilityai-stable-diffusion-2-1-base-smashed --local-dir stabilityai-stable-diffusion-2-1-base-smashed --local-dir-use-symlinks False
     ```
    Alternatively, you can download them manually.
 3. Loading the model.
@@ -40,7 +40,7 @@ You can run the smashed model by:
 You can achieve this by running the following code:
 ```python
 from pruna_engine.PrunaModel import PrunaModel  # Step (1): install and import `pruna-engine` package.
-model_path = "stabilityai-stable-diffusion-2-1-base-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=768, image_width=768)[0]  # Step (4): run the model.
 ```

 2. Downloading the model files. This can be done using the Hugging Face CLI with the following commands:
     ```bash
     mkdir stabilityai-stable-diffusion-2-1-base-smashed
+    huggingface-cli download PrunaAI/stabilityai-stable-diffusion-2-1-base-turbo-green-smashed --local-dir stabilityai-stable-diffusion-2-1-base-turbo-green-smashed --local-dir-use-symlinks False
     ```
    Alternatively, you can download them manually.
 3. Loading the model.
 You can achieve this by running the following code:
 ```python
 from pruna_engine.PrunaModel import PrunaModel  # Step (1): install and import `pruna-engine` package.
+model_path = "stabilityai-stable-diffusion-2-1-base-turbo-green-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=768, image_width=768)[0]  # Step (4): run the model.
 ```