johnrachwanpruna
commited on
Commit
•
c985f91
1
Parent(s):
eed4d4f
Update README.md
Browse files
README.md
CHANGED
@@ -39,10 +39,9 @@ You can achieve this by running the following code:
|
|
39 |
from transformers.utils.hub import cached_file
|
40 |
from pruna_engine.PrunaModel import PrunaModel # Step (1): install and import `pruna-engine` package.
|
41 |
|
42 |
-
|
43 |
-
model_path = cached_file("PrunaAI/REPO", "model") # Step (2): download the model files at `model_path`.
|
44 |
smashed_model = PrunaModel.load_model(model_path) # Step (3): load the model.
|
45 |
-
|
46 |
```
|
47 |
|
48 |
## Configurations
|
|
|
39 |
from transformers.utils.hub import cached_file
|
40 |
from pruna_engine.PrunaModel import PrunaModel # Step (1): install and import `pruna-engine` package.
|
41 |
|
42 |
+
model_path = cached_file("PrunaAI/SimianLuo-LCM_Dreamshaper_v7-smashed", "model") # Step (2): download the model files at `model_path`.
|
|
|
43 |
smashed_model = PrunaModel.load_model(model_path) # Step (3): load the model.
|
44 |
+
smashed_model("Self-portrait oil painting, a beautiful cyborg with golden hair, 8k", num_inference_steps=4)[0] # Step (4): run the model.
|
45 |
```
|
46 |
|
47 |
## Configurations
|