Index

_

__call__() (PrunaModel method)

__init__() (PrunaDataModule method) (PrunaModel method)

A

add() (SmashConfig method) add_data() (SmashConfig method) add_processor() (SmashConfig method)

add_target_module() (SmashConfig method) add_tokenizer() (SmashConfig method) allow_zero_length_dataloader_with_multiple_devices (PrunaDataModule attribute) audio_collate() (in module pruna.data.collate)

C

cleanup_cache_dir() (SmashConfig method)

compute_stateful_metrics() (EvaluationAgent method) compute_stateless_metrics() (EvaluationAgent method)

D

destroy() (PrunaModel method)

DiffuserHandler (class in pruna.engine.handler.handler_diffuser) disable_saving() (SmashConfig method)

E

evaluate() (EvaluationAgent method)

EvaluationAgent (class in pruna.evaluation.evaluation_agent)

F

flush_configuration() (SmashConfig method) from_datasets() (PrunaDataModule class method) from_dict() (SmashConfig class method)

from_list() (SmashConfig class method) from_pretrained() (PrunaModel static method) from_string() (PrunaDataModule class method)

G

get_active_algorithms() (SmashConfig method) get_nn_modules() (PrunaModel method)

get_pairwise_stateful_metrics() (Task method) get_single_stateful_metrics() (Task method) get_stateless_metrics() (Task method)

I

image_classification_collate() (in module pruna.data.collate) image_generation_collate() (in module pruna.data.collate)

is_batch_size_locked() (SmashConfig method) is_instance() (PrunaModel method) is_pairwise_evaluation() (Task method)

L

limit_datasets() (PrunaDataModule method) load_from_json() (SmashConfig method) lock_batch_size() (SmashConfig method)

log_model_info() (DiffuserHandler method) (StandardHandler method) (TransformerHandler method)

O

overwrite_algorithm_order() (SmashConfig method)

P

prepare_data_per_node (PrunaDataModule attribute) prepare_inputs() (DiffuserHandler method) (StandardHandler method) (TransformerHandler method) prepare_model() (EvaluationAgent method)

process_output() (DiffuserHandler method) (StandardHandler method) (TransformerHandler method) PrunaDataModule (class in pruna.data.pruna_datamodule) PrunaModel (class in pruna.engine.pruna_model) push_to_hub() (PrunaModel method)

Q

question_answering_collate() (in module pruna.data.collate)

R

reset_cache_dir() (SmashConfig method)

run_inference() (PrunaModel method)

S

save_pretrained() (PrunaModel method) save_to_json() (SmashConfig method) set_telemetry_metrics() (in module pruna.telemetry)

set_to_eval() (PrunaModel method) smash() (in module pruna.smash) SmashConfig (class in pruna.config.smash_config) StandardHandler (class in pruna.engine.handler.handler_standard)

T

Task (class in pruna.evaluation.task) test_dataloader() (PrunaDataModule method) (SmashConfig method)

text_generation_collate() (in module pruna.data.collate) train_dataloader() (PrunaDataModule method) (SmashConfig method) TransformerHandler (class in pruna.engine.handler.handler_transformer)

U

update_stateful_metrics() (EvaluationAgent method)

V

val_dataloader() (PrunaDataModule method) (SmashConfig method)