|
|
--- |
|
|
library_name: diffusers |
|
|
tags: |
|
|
- pruna-ai |
|
|
--- |
|
|
|
|
|
# Model Card for davidberenstein1957/stable-diffusion-v1-4-smashed-1 |
|
|
|
|
|
This model was created using the [pruna](https://github.com/PrunaAI/pruna) library. Pruna is a model optimization framework built for developers, enabling you to deliver more efficient models with minimal implementation overhead. |
|
|
|
|
|
## Usage |
|
|
|
|
|
First things first, you need to install the pruna library: |
|
|
|
|
|
```bash |
|
|
pip install "pruna[full]" |
|
|
``` |
|
|
|
|
|
You can then load this model using the following code: |
|
|
|
|
|
```python |
|
|
from pruna import PrunaModel |
|
|
|
|
|
loaded_model = PrunaModel.from_hub( |
|
|
"davidberenstein1957/stable-diffusion-v1-4-smashed-1" |
|
|
) |
|
|
``` |
|
|
|
|
|
After loading the model, you can use the inference methods of the original model. |
|
|
|
|
|
## Smash Configuration |
|
|
|
|
|
The compression configuration of the model is stored in the `smash_config.json` file. |
|
|
|
|
|
```bash |
|
|
{ |
|
|
"batcher": null, |
|
|
"cacher": "deepcache", |
|
|
"compiler": null, |
|
|
"pruner": null, |
|
|
"quantizer": null, |
|
|
"deepcache_interval": 2, |
|
|
"max_batch_size": 1, |
|
|
"device": "cuda", |
|
|
"save_fns": [], |
|
|
"load_fns": [ |
|
|
"diffusers" |
|
|
], |
|
|
"reapply_after_load": { |
|
|
"pruner": null, |
|
|
"quantizer": null, |
|
|
"cacher": "deepcache", |
|
|
"compiler": null, |
|
|
"batcher": null |
|
|
} |
|
|
} |
|
|
``` |
|
|
|
|
|
## Model Configuration |
|
|
|
|
|
The configuration of the model is stored in the `*.json` files. |
|
|
|
|
|
```bash |
|
|
{ |
|
|
"model_index": { |
|
|
"_class_name": "StableDiffusionPipeline", |
|
|
"_diffusers_version": "0.33.1", |
|
|
"_name_or_path": "CompVis/stable-diffusion-v1-4", |
|
|
"feature_extractor": [ |
|
|
"transformers", |
|
|
"CLIPImageProcessor" |
|
|
], |
|
|
"image_encoder": [ |
|
|
null, |
|
|
null |
|
|
], |
|
|
"requires_safety_checker": true, |
|
|
"safety_checker": [ |
|
|
"stable_diffusion", |
|
|
"StableDiffusionSafetyChecker" |
|
|
], |
|
|
"scheduler": [ |
|
|
"diffusers", |
|
|
"PNDMScheduler" |
|
|
], |
|
|
"text_encoder": [ |
|
|
"transformers", |
|
|
"CLIPTextModel" |
|
|
], |
|
|
"tokenizer": [ |
|
|
"transformers", |
|
|
"CLIPTokenizer" |
|
|
], |
|
|
"unet": [ |
|
|
"diffusers", |
|
|
"UNet2DConditionModel" |
|
|
], |
|
|
"vae": [ |
|
|
"diffusers", |
|
|
"AutoencoderKL" |
|
|
] |
|
|
} |
|
|
} |
|
|
``` |
|
|
|
|
|
## ๐ Join the Pruna AI community! |
|
|
|
|
|
[](https://twitter.com/PrunaAI) |
|
|
[](https://github.com/PrunaAI) |
|
|
[](https://www.linkedin.com/company/93832878/admin/feed/posts/?feedType=following) |
|
|
[](https://discord.com/invite/rskEr4BZJx) |
|
|
[](https://www.reddit.com/r/PrunaAI/) |