Upload folder using huggingface_hub (#4)
Browse files- 67d3efd3bca75bd7389065a528e7cc4aa8a67b4c6317a5bc40ba498a4fc9adc7 (1d8978450fd66b27e83a99e3443ae162da4af808)
- 643bf0a5f581099397d8ae86057cd0422451dc8a2912cb731f3c5cbb5fb4f433 (91ec119ab59a95e68abae84889d8d2ea67e4ce80)
- README.md +1 -1
- config.json +1 -1
- model/optimized_model.pkl +2 -2
- model/smash_config.json +2 -2
- plots.png +0 -0
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',
|
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=1024, width=1024)[0][0] # Run the model where x is the expected input of.
|
70 |
```
|
71 |
|
72 |
## Configurations
|
config.json
CHANGED
@@ -1 +1 @@
|
|
1 |
-
{"pruners": "None", "pruning_ratio": 0.0, "factorizers": "None", "quantizers": "None", "n_quantization_bits": 32, "output_deviation": 0.
|
|
|
1 |
+
{"pruners": "None", "pruning_ratio": 0.0, "factorizers": "None", "quantizers": "None", "n_quantization_bits": 32, "output_deviation": 0.005, "compilers": "['diffusers2']", "static_batch": true, "static_shape": false, "controlnet": "None", "unet_dim": 4, "device": "cuda", "save_dir": "/ceph/hdd/staff/charpent/models/.models/optimized_model", "batch_size": 1, "max_batch_size": 1, "image_height": 512, "image_width": 512, "version": "1.5", "task": "txt2img", "model_name": "runwayml/stable-diffusion-v1-5", "weight_name": "None", "save_load_fn": "stable_fast"}
|
model/optimized_model.pkl
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1788eda1cb4fa185ceed619239c39e9d561a399e2e2e3c81e714c86496e08f24
|
3 |
+
size 2743389116
|
model/smash_config.json
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ac30e3ad627076c9785b1719c6577985572ccbaa7e30442fa95b86be71854c84
|
3 |
+
size 743
|
plots.png
CHANGED