Index

_

__call__() (PrunaModel method) __init__() (OptimizationAgent method) (PrunaDataModule method) (PrunaModel method)

A

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

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

D

destroy() (PrunaModel method)

DiffuserHandler (class in pruna.engine.handler.handler_diffuser)

E

evaluate() (EvaluationAgent method)

EvaluationAgent (class in pruna.evaluation.evaluation_agent)

F

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

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

G

get_nn_modules() (PrunaModel method)

I

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

image_generation_collate() (in module pruna.data.collate) instant_search() (OptimizationAgent method)

L

limit_datasets() (PrunaDataModule method)

load_dict() (SmashConfig method) load_from_json() (SmashConfig method)

M

move_to_device() (PrunaModel method)

O

OptimizationAgent (class in pruna_pro.optimization_agent)

P

prepare_data_per_node (PrunaDataModule attribute) probabilistic_search() (OptimizationAgent method)

PrunaDataModule (class in pruna.data.pruna_datamodule) PrunaModel (class in pruna.engine.pruna_model)

Q

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

R

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)

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

V

val_dataloader() (PrunaDataModule method)