Update README.md
Browse files
README.md
CHANGED
|
@@ -32,7 +32,7 @@ You can run the smashed model by:
|
|
| 32 |
2. Downloading the model files. This can be done using the Hugging Face CLI with the following commands:
|
| 33 |
```bash
|
| 34 |
mkdir CompVis-stable-diffusion-v1-4-smashed
|
| 35 |
-
huggingface-cli download PrunaAI/CompVis-stable-diffusion-v1-4-smashed --local-dir CompVis-stable-diffusion-v1-4-smashed --local-dir-use-symlinks False
|
| 36 |
```
|
| 37 |
Alternatively, you can download them manually.
|
| 38 |
3. Loading the model.
|
|
@@ -41,7 +41,7 @@ You can achieve this by running the following code:
|
|
| 41 |
|
| 42 |
```python
|
| 43 |
from pruna_engine.PrunaModel import PrunaModel # Step (1): install and import `pruna-engine` package.
|
| 44 |
-
model_path = "CompVis-stable-diffusion-v1-4-smashed/model" # Step (2): specify the downloaded model path.
|
| 45 |
smashed_model = PrunaModel.load_model(model_path) # Step (3): load the model.
|
| 46 |
y = smashed_model(prompt="a silly prune with a face in high definition", image_height=512, image_width=512)[0] # Step (4): run the model.
|
| 47 |
```
|
|
|
|
| 32 |
2. Downloading the model files. This can be done using the Hugging Face CLI with the following commands:
|
| 33 |
```bash
|
| 34 |
mkdir CompVis-stable-diffusion-v1-4-smashed
|
| 35 |
+
huggingface-cli download PrunaAI/CompVis-stable-diffusion-v1-4-turbo-tiny-green-smashed --local-dir CompVis-stable-diffusion-v1-4-turbo-tiny-green-smashed --local-dir-use-symlinks False
|
| 36 |
```
|
| 37 |
Alternatively, you can download them manually.
|
| 38 |
3. Loading the model.
|
|
|
|
| 41 |
|
| 42 |
```python
|
| 43 |
from pruna_engine.PrunaModel import PrunaModel # Step (1): install and import `pruna-engine` package.
|
| 44 |
+
model_path = "CompVis-stable-diffusion-v1-4-turbo-tiny-green-smashed/model" # Step (2): specify the downloaded model path.
|
| 45 |
smashed_model = PrunaModel.load_model(model_path) # Step (3): load the model.
|
| 46 |
y = smashed_model(prompt="a silly prune with a face in high definition", image_height=512, image_width=512)[0] # Step (4): run the model.
|
| 47 |
```
|