Update README.md
Browse files
README.md
CHANGED
@@ -33,7 +33,7 @@ You can run the smashed model by:
|
|
33 |
2. Downloading the model files. This can be done using the Hugging Face CLI with the following commands:
|
34 |
```bash
|
35 |
mkdir SimianLuo-LCM_Dreamshaper_v7-smashed
|
36 |
-
huggingface-cli download PrunaAI/SimianLuo-LCM_Dreamshaper_v7-smashed --local-dir SimianLuo-LCM_Dreamshaper_v7-smashed --local-dir-use-symlinks False
|
37 |
```
|
38 |
Alternatively, you can download them manually.
|
39 |
3. Loading the model.
|
@@ -44,7 +44,7 @@ You can achieve this by running the following code:
|
|
44 |
```python
|
45 |
from pruna_engine.PrunaModel import PrunaModel # Step (1): install and import `pruna-engine` package.
|
46 |
|
47 |
-
model_path = "SimianLuo-LCM_Dreamshaper_v7-smashed/model" # Step (2): specify the downloaded model path.
|
48 |
smashed_model = PrunaModel.load_model(model_path) # Step (3): load the model.
|
49 |
smashed_model("Self-portrait oil painting, a beautiful cyborg with golden hair, 8k", num_inference_steps=4)[0] # Step (4): run the model.
|
50 |
```
|
|
|
33 |
2. Downloading the model files. This can be done using the Hugging Face CLI with the following commands:
|
34 |
```bash
|
35 |
mkdir SimianLuo-LCM_Dreamshaper_v7-smashed
|
36 |
+
huggingface-cli download PrunaAI/SimianLuo-LCM_Dreamshaper_v7-turbo-tiny-green-smashed --local-dir SimianLuo-LCM_Dreamshaper_v7-turbo-tiny-green-smashed --local-dir-use-symlinks False
|
37 |
```
|
38 |
Alternatively, you can download them manually.
|
39 |
3. Loading the model.
|
|
|
44 |
```python
|
45 |
from pruna_engine.PrunaModel import PrunaModel # Step (1): install and import `pruna-engine` package.
|
46 |
|
47 |
+
model_path = "SimianLuo-LCM_Dreamshaper_v7-turbo-tiny-green-smashed/model" # Step (2): specify the downloaded model path.
|
48 |
smashed_model = PrunaModel.load_model(model_path) # Step (3): load the model.
|
49 |
smashed_model("Self-portrait oil painting, a beautiful cyborg with golden hair, 8k", num_inference_steps=4)[0] # Step (4): run the model.
|
50 |
```
|