This model was created using the 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:

pip install pruna_pro

You can use the transformers library to load the model 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_pro import PrunaProModel

loaded_model = PrunaProModel.from_pretrained(
    "loulou2/tiny_llama_higgs"
)
# we can then run inference using the methods supported by the base model

Alternatively, you can visit the Pruna documentation for more 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,
    "distributer": null,
    "enhancer": null,
    "factorizer": null,
    "kernel": null,
    "pruner": null,
    "quantizer": "higgs",
    "recoverer": null,
    "higgs_group_size": 256,
    "higgs_hadamard_size": 1024,
    "higgs_p": 2,
    "higgs_weight_bits": 4,
    "batch_size": 1,
    "device": "cuda",
    "device_map": null,
    "save_fns": [
        "transformers_higgs"
    ],
    "load_fns": [
        "transformers_higgs"
    ],
    "reapply_after_load": {
        "factorizer": null,
        "pruner": null,
        "quantizer": null,
        "distiller": null,
        "kernel": null,
        "cacher": null,
        "recoverer": null,
        "distributer": null,
        "compiler": null,
        "batcher": null,
        "enhancer": null
    }
}

🌍 Join the Pruna AI community!

Discord

pruna-test/tiny_llama_higgs

Model Card for loulou2/tiny_llama_higgs

Usage

Smash Configuration

🌍 Join the Pruna AI community!

pruna-test/
tiny_llama_higgs