Views
No views yet
pip install "pruna[full]"1from pruna import PrunaModel
2
3loaded_model = PrunaModel.from_hub(
4 "davidberenstein1957/stable-diffusion-v1-4-smashed"
5)smash_config.json file.1{
2 "batcher": null,
3 "cacher": "deepcache",
4 "compiler": null,
5 "pruner": null,
6 "quantizer": null,
7 "deepcache_interval": 2,
8 "max_batch_size": 1,
9 "device": "cuda",
10 "save_fns": [],
11 "load_fns": [
12 "diffusers"
13 ],
14 "reapply_after_load": {
15 "pruner": null,
16 "quantizer": null,
17 "cacher": "deepcache",
18 "compiler": null,
19 "batcher": null
20 }
21}*.json files.1{
2 "model_index": {
3 "_class_name": "StableDiffusionPipeline",
4 "_diffusers_version": "0.33.1",
5 "_name_or_path": "CompVis/stable-diffusion-v1-4",
6 "feature_extractor": [
7 "transformers",
8 "CLIPImageProcessor"
9 ],
10 "image_encoder": [
11 null,
12 null
13 ],
14 "requires_safety_checker": true,
15 "safety_checker": [
16 "stable_diffusion",
17 "StableDiffusionSafetyChecker"
18 ],
19 "scheduler": [
20 "diffusers",
21 "PNDMScheduler"
22 ],
23 "text_encoder": [
24 "transformers",
25 "CLIPTextModel"
26 ],
27 "tokenizer": [
28 "transformers",
29 "CLIPTokenizer"
30 ],
31 "unet": [
32 "diffusers",
33 "UNet2DConditionModel"
34 ],
35 "vae": [
36 "diffusers",
37 "AutoencoderKL"
38 ]
39 }
40}