fedem

• fedem package
  • Subpackages
    • fedem.client package
      • Module contents
    • fedem.configurations package
      • Submodules
      • fedem.configurations.mamba module
      • Module contents
    • fedem.models package
      • Submodules
      • fedem.models.mamba module
      • Module contents
    • fedem.server package
      • Module contents
    • fedem.trainer package
      • Module contents
    • fedem.utils package
      • Submodules
      • fedem.utils.huggingface module
      • Module contents
  • Module contents
• fedem.client package
  • Module contents
    • Seshu
      • Seshu.push_to_hub()
      • Seshu.tokenize()
      • Seshu.train_lora()
• fedem.server package
  • Module contents
    • MambaTrainer
      • MambaTrainer.compute_loss()
    • Seshu
      • Seshu.model_merge_eval()
      • Seshu.pretrain()
      • Seshu.tokenize()
    • compute_loss()
    • create_JSON()
    • evaluation()
    • get_checkpoint_model()
    • get_data()
    • load_data()
    • load_json()
    • make_config()
    • model_merge()
    • print_trainable_parameters()
• fedem.configurations package
  • Submodules
  • fedem.configurations.mamba module
    • MambaConfig
      • MambaConfig.model_type
  • Module contents
• fedem.models package
  • Submodules
  • fedem.models.mamba module
    • MambaBlock
      • MambaBlock.forward()
      • MambaBlock.selective_scan()
      • MambaBlock.ssm()
    • MambaForCausalLM
      • MambaForCausalLM.forward()
      • MambaForCausalLM.get_decoder()
      • MambaForCausalLM.get_input_embeddings()
      • MambaForCausalLM.get_output_embeddings()
      • MambaForCausalLM.prepare_inputs_for_generation()
      • MambaForCausalLM.set_decoder()
      • MambaForCausalLM.set_input_embeddings()
      • MambaForCausalLM.set_output_embeddings()
    • MambaModel
      • MambaModel.forward()
      • MambaModel.get_input_embeddings()
      • MambaModel.set_input_embeddings()
    • MambaPreTrainedModel
      • MambaPreTrainedModel.base_model_prefix
      • MambaPreTrainedModel.config_class
      • MambaPreTrainedModel.supports_gradient_checkpointing
    • MambaRMSNorm
      • MambaRMSNorm.forward()
  • Module contents
• fedem.trainer package
  • Module contents
    • MambaTrainer
      • MambaTrainer.compute_loss()
• fedem.utils package
  • Submodules
  • fedem.utils.huggingface module
    • get_client_details()
    • verify_user_with_org()
  • Module contents
    • get_checkpoint_model()
    • load_data()
    • load_json()
    • load_model()
    • load_model_pretrained()
    • load_model_with_LoRA()
    • load_tokenizer()
    • make_config()
    • print_trainable_parameters()
    • split_data()