Views
No views yet
pip install src1from src import PrunaProModel
2
3loaded_model = PrunaProModel.from_hub(
4 "PrunaAI/test-load-tiny-random-llama3-smashed-pro"
5)smash_config.json file, which describes the optimization methods that were applied to the model.1{
2 "batcher": null,
3 "cacher": null,
4 "compiler": null,
5 "distiller": null,
6 "distributer": null,
7 "enhancer": null,
8 "factorizer": null,
9 "pruner": null,
10 "quantizer": null,
11 "recoverer": null,
12 "batch_size": 1,
13 "device": "cpu",
14 "device_map": null,
15 "save_fns": [],
16 "load_fns": [
17 "transformers"
18 ],
19 "reapply_after_load": {
20 "factorizer": null,
21 "pruner": null,
22 "quantizer": null,
23 "distiller": null,
24 "cacher": null,
25 "recoverer": null,
26 "distributer": null,
27 "compiler": null,
28 "batcher": null,
29 "enhancer": null
30 }
31}