86768fcb33c459e45831c07a638bd4cbad9c23b4d14e9a6335178218dae92195
Browse files
README.md
CHANGED
@@ -63,9 +63,9 @@ You can run the smashed model with these steps:
|
|
63 |
```python
|
64 |
from pruna_engine.PrunaModel import PrunaModel
|
65 |
|
66 |
-
model_path = "
|
67 |
smashed_model = PrunaModel.load_model(model_path) # Load the model.
|
68 |
-
|
69 |
```
|
70 |
|
71 |
## Configurations
|
|
|
63 |
```python
|
64 |
from pruna_engine.PrunaModel import PrunaModel
|
65 |
|
66 |
+
model_path = "runwayml-stable-diffusion-v1-5-turbo-tiny-green-smashed/model" # Specify the downloaded model path.
|
67 |
smashed_model = PrunaModel.load_model(model_path) # Load the model.
|
68 |
+
smashed_model(prompt='Beautiful fruits in trees', image_height=1024, image_width=1024)[0][0] # Run the model where x is the expected input of.
|
69 |
```
|
70 |
|
71 |
## Configurations
|
plots.png
CHANGED