| --- |
| library_name: diffusers |
| tags: |
| - pruna-ai |
| --- |
| |
| # Model Card for PrunaAI/test-load-tiny-stable-diffusion-pipe-smashed-pro |
|
|
| 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 |
| ``` |
|
|
| You can [use the diffusers library to load the model](https://huggingface.co/PrunaAI/test-load-tiny-stable-diffusion-pipe-smashed-pro?library=diffusers) but this might not include all optimizations by default. |
|
|
| To ensure that all optimizations are applied, use the pruna library to load the model using the following code: |
|
|
| ```python |
| from pruna import PrunaModel |
| |
| loaded_model = PrunaModel.from_hub( |
| "PrunaAI/test-load-tiny-stable-diffusion-pipe-smashed-pro" |
| ) |
| ``` |
|
|
| After loading the model, you can use the inference methods of the original model. Take a look at the [documentation](https://pruna.readthedocs.io/en/latest/index.html) for more usage information. |
|
|
| ## Smash Configuration |
|
|
| The compression configuration of the model is stored in the `smash_config.json` file, which describes the optimization methods that were applied to the model. |
|
|
| ```bash |
| { |
| "batcher": null, |
| "cacher": null, |
| "compiler": null, |
| "distiller": null, |
| "enhancer": null, |
| "factorizer": null, |
| "pruner": null, |
| "quantizer": null, |
| "recoverer": null, |
| "batch_size": 1, |
| "device": "cpu", |
| "save_fns": [], |
| "load_fns": [ |
| "diffusers" |
| ], |
| "reapply_after_load": { |
| "factorizer": null, |
| "pruner": null, |
| "quantizer": null, |
| "distiller": null, |
| "cacher": null, |
| "recoverer": null, |
| "compiler": null, |
| "batcher": null, |
| "enhancer": null |
| } |
| } |
| ``` |
|
|
| ## 🌍 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/) |