sharpenb commited on
Commit
41a8165
1 Parent(s): 200aff3

Upload folder using huggingface_hub (#3)

Browse files

- 8238376fbf1a0e045a79cd7bfc61d20668072182a6937a8a294489cda8ac7a54 (a4463eeec5f83a9880cd4a1371fac4981f442cc2)
- 86768fcb33c459e45831c07a638bd4cbad9c23b4d14e9a6335178218dae92195 (eda4e01519748f21fe2f8714941a4cff7936cef3)

Files changed (3) hide show
  1. README.md +2 -2
  2. model/optimized_model.pkl +2 -2
  3. plots.png +0 -0
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 = "PrunaAI/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
- y = smashed_model(x) # Run the model where x is the expected input of.
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
model/optimized_model.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:32f59a4e05e4485e7bb0f5ff093fca7256dc823c5db956cbbbeeba43f15ca2e1
3
- size 2743389214
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:838dcdb516f2924e875de850ebc8480a694976a6fef534a11a7201ddd370fac5
3
+ size 2743389078
plots.png CHANGED