Views
No views yet
pip install pruna_pro1from pruna_pro import PrunaProModel
2
3loaded_model = PrunaProModel.from_pretrained(
4 "pruna-test/test-save-tiny-random-llama3-smashed-pro"
5)
6# we can then run inference using the methods supported by the base modelsmash_config.json file, which describes the optimization methods that were applied to the model.1{
2 "adaptive": false,
3 "auto": false,
4 "awq": false,
5 "bottleneck": false,
6 "c_generate": false,
7 "c_translate": false,
8 "c_whisper": false,
9 "custom_schedule": false,
10 "deepcache": false,
11 "diffusers_higgs": false,
12 "diffusers_int8": false,
13 "fastercache": false,
14 "flash_attn3": false,
15 "flux_caching": false,
16 "fora": false,
17 "fp4": false,
18 "fp8": false,
19 "gptq": false,
20 "half": false,
21 "higgs": false,
22 "hqq": false,
23 "hqq_diffusers": false,
24 "hyper": false,
25 "ifw": false,
26 "img2img_denoise": false,
27 "kvpress": false,
28 "llama_cpp": false,
29 "llm_int8": false,
30 "moe_kernel_tuner": false,
31 "pab": false,
32 "padding_pruning": false,
33 "periodic": false,
34 "prores": false,
35 "qkv_diffusers": false,
36 "quanto": false,
37 "radial_attn": false,
38 "realesrgan_upscale": false,
39 "reduce_noe": false,
40 "ring_attn": false,
41 "sage_attn": false,
42 "stable_fast": false,
43 "static_fp8_diffusers": false,
44 "taylor": false,
45 "taylor_auto": false,
46 "text_to_image_distillation_inplace_perp": false,
47 "text_to_image_distillation_lora": false,
48 "text_to_image_distillation_perp": false,
49 "text_to_image_inplace_perp": false,
50 "text_to_image_lora": false,
51 "text_to_image_perp": false,
52 "text_to_text_inplace_perp": false,
53 "text_to_text_lora": false,
54 "text_to_text_perp": false,
55 "time_aware_fp8_diffusers": false,
56 "token_merging": false,
57 "torch_compile": false,
58 "torch_dynamic": false,
59 "torch_structured": false,
60 "torch_unstructured": false,
61 "torchao": false,
62 "torchao_autoquant": false,
63 "x_fast": false,
64 "zipar": false,
65 "batch_size": 1,
66 "device": "cpu",
67 "device_map": null,
68 "save_fns": [],
69 "save_artifacts_fns": [],
70 "load_fns": [
71 "transformers"
72 ],
73 "load_artifacts_fns": [],
74 "reapply_after_load": {}
75}