sharpenb commited on
Commit
8b49a33
1 Parent(s): 2b5f896

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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=1024, width=1024)[0][0] # Run the model where x is the expected input of.
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