Update README.md
Browse files
README.md
CHANGED
|
@@ -66,7 +66,7 @@ You can run the smashed model with these steps:
|
|
| 66 |
|
| 67 |
model_path = "runwayml-stable-diffusion-v1-5-turbo-tiny-green-smashed/model" # Specify the downloaded model path.
|
| 68 |
smashed_model = PrunaModel.load_model(model_path) # Load the model.
|
| 69 |
-
smashed_model(prompt='Beautiful fruits in trees', height=
|
| 70 |
```
|
| 71 |
|
| 72 |
## Configurations
|
|
|
|
| 66 |
|
| 67 |
model_path = "runwayml-stable-diffusion-v1-5-turbo-tiny-green-smashed/model" # Specify the downloaded model path.
|
| 68 |
smashed_model = PrunaModel.load_model(model_path) # Load the model.
|
| 69 |
+
smashed_model(prompt='Beautiful fruits in trees', height=512, width=512)[0][0] # Run the model where x is the expected input of.
|
| 70 |
```
|
| 71 |
|
| 72 |
## Configurations
|