davidberenstein1957's picture
Add files using upload-large-folder tool
b419e9e verified
---
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!
[![Twitter](https://img.shields.io/twitter/follow/PrunaAI?style=social)](https://twitter.com/PrunaAI)
[![GitHub](https://img.shields.io/github/followers/PrunaAI?label=Follow%20%40PrunaAI&style=social)](https://github.com/PrunaAI)
[![LinkedIn](https://img.shields.io/badge/LinkedIn-Connect-blue)](https://www.linkedin.com/company/93832878/admin/feed/posts/?feedType=following)
[![Discord](https://img.shields.io/badge/Discord-Join%20Us-blue?style=social&logo=discord)](https://discord.com/invite/rskEr4BZJx)
[![Reddit](https://img.shields.io/reddit/subreddit-subscribers/PrunaAI?style=social)](https://www.reddit.com/r/PrunaAI/)