Index

B

base_model_prefix (fedem.models.mamba.MambaPreTrainedModel attribute)

C

compute_loss() (fedem.server.MambaTrainer method) (fedem.trainer.MambaTrainer method) (in module fedem.server)

config_class (fedem.models.mamba.MambaPreTrainedModel attribute) create_JSON() (in module fedem.server)

E

evaluation() (in module fedem.server)

F

fedem module fedem.client module fedem.configurations module fedem.configurations.mamba module fedem.models module fedem.models.mamba module

fedem.server module fedem.trainer module fedem.utils module fedem.utils.huggingface module forward() (fedem.models.mamba.MambaBlock method) (fedem.models.mamba.MambaForCausalLM method) (fedem.models.mamba.MambaModel method) (fedem.models.mamba.MambaRMSNorm method)

G

get_checkpoint_model() (in module fedem.server) (in module fedem.utils) get_client_details() (in module fedem.utils.huggingface) get_data() (in module fedem.server)

get_decoder() (fedem.models.mamba.MambaForCausalLM method) get_input_embeddings() (fedem.models.mamba.MambaForCausalLM method) (fedem.models.mamba.MambaModel method) get_output_embeddings() (fedem.models.mamba.MambaForCausalLM method)

L

load_data() (in module fedem.server) (in module fedem.utils) load_json() (in module fedem.server) (in module fedem.utils)

load_model() (in module fedem.utils) load_model_pretrained() (in module fedem.utils) load_model_with_LoRA() (in module fedem.utils) load_tokenizer() (in module fedem.utils)

M

make_config() (in module fedem.server) (in module fedem.utils) MambaBlock (class in fedem.models.mamba) MambaConfig (class in fedem.configurations.mamba) MambaForCausalLM (class in fedem.models.mamba) MambaModel (class in fedem.models.mamba) MambaPreTrainedModel (class in fedem.models.mamba) MambaRMSNorm (class in fedem.models.mamba) MambaTrainer (class in fedem.server) (class in fedem.trainer) model_merge() (in module fedem.server) model_merge_eval() (fedem.server.Seshu method)

model_type (fedem.configurations.mamba.MambaConfig attribute) module fedem fedem.client fedem.configurations fedem.configurations.mamba fedem.models fedem.models.mamba fedem.server fedem.trainer fedem.utils fedem.utils.huggingface

P

prepare_inputs_for_generation() (fedem.models.mamba.MambaForCausalLM method) pretrain() (fedem.server.Seshu method)

print_trainable_parameters() (in module fedem.server) (in module fedem.utils) push_to_hub() (fedem.client.Seshu method)

S

selective_scan() (fedem.models.mamba.MambaBlock method) Seshu (class in fedem.client) (class in fedem.server) set_decoder() (fedem.models.mamba.MambaForCausalLM method) set_input_embeddings() (fedem.models.mamba.MambaForCausalLM method) (fedem.models.mamba.MambaModel method)

set_output_embeddings() (fedem.models.mamba.MambaForCausalLM method) split_data() (in module fedem.utils) ssm() (fedem.models.mamba.MambaBlock method) supports_gradient_checkpointing (fedem.models.mamba.MambaPreTrainedModel attribute)

T

tokenize() (fedem.client.Seshu method) (fedem.server.Seshu method)

train_lora() (fedem.client.Seshu method)

V

verify_user_with_org() (in module fedem.utils.huggingface)