| | --- |
| | 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/) |