Update README.md
Browse files
README.md
CHANGED
@@ -42,7 +42,7 @@ from pruna_engine.PrunaModel import PrunaModel # Step (1): install and import `
|
|
42 |
...
|
43 |
model_path = cached_file("PrunaAI/stabilityai-stable-diffusion-xl-base-1.0-smashed", "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(prompt="a
|
46 |
```
|
47 |
|
48 |
## Configurations
|
|
|
42 |
...
|
43 |
model_path = cached_file("PrunaAI/stabilityai-stable-diffusion-xl-base-1.0-smashed", "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(prompt="a silly prune with a face in high definition", image_height=1024, image_width=1024)[0] # Step (4): run the model.
|
46 |
```
|
47 |
|
48 |
## Configurations
|