johnrachwanpruna commited on
Commit
c985f91
1 Parent(s): eed4d4f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
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
- y = smashed_model(x) # Step (4): run the model.
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