| | --- |
| | library_name: diffusers |
| | tags: |
| | - pruna-ai |
| | --- |
| | |
| | # Model Card for davidberenstein1957/stable-diffusion-v1-4-smashed |
| |
|
| | 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" |
| | ) |
| | ``` |
| |
|
| | 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/) |