lmflow

Subpackages

• lmflow.datasets
  • lmflow.datasets.dataset
  • lmflow.datasets.multi_modal_dataset
• lmflow.models
  • lmflow.models.interfaces
    • lmflow.models.interfaces.tunable
  • lmflow.models.vision_encoder
    • lmflow.models.vision_encoder.clip_encoder
  • lmflow.models.auto_model
  • lmflow.models.base_model
  • lmflow.models.decoder_model
  • lmflow.models.encoder_decoder_model
  • lmflow.models.hf_decoder_model
  • lmflow.models.hf_encoder_decoder_model
  • lmflow.models.regression_model
  • lmflow.models.text_regression_model
  • lmflow.models.vision2seq_model
• lmflow.pipeline
  • lmflow.pipeline.utils
    • lmflow.pipeline.utils.peft_trainer
    • lmflow.pipeline.utils.raft_trainer
  • lmflow.pipeline.auto_pipeline
  • lmflow.pipeline.base_aligner
  • lmflow.pipeline.base_pipeline
  • lmflow.pipeline.base_tuner
  • lmflow.pipeline.dpo_aligner
  • lmflow.pipeline.evaluator
  • lmflow.pipeline.finetuner
  • lmflow.pipeline.inferencer
  • lmflow.pipeline.raft_aligner
• lmflow.utils
  • lmflow.utils.conversation_template
    • lmflow.utils.conversation_template.base
    • lmflow.utils.conversation_template.chatml
    • lmflow.utils.conversation_template.deepseek
    • lmflow.utils.conversation_template.llama
    • lmflow.utils.conversation_template.phi
    • lmflow.utils.conversation_template.qwen
  • lmflow.utils.flash_attention
    • lmflow.utils.flash_attention.bloom_flash_attention
    • lmflow.utils.flash_attention.gpt2_flash_attention
    • lmflow.utils.flash_attention.gpt_neo_flash_attention
    • lmflow.utils.flash_attention.llama_flash_attention
    • lmflow.utils.flash_attention.triton_flash_attention
  • lmflow.utils.position_interpolation
    • lmflow.utils.position_interpolation.llama_rope_scaled_monkey_patch
  • lmflow.utils.constants
  • lmflow.utils.data_utils
  • lmflow.utils.llava_conversation_lib
  • lmflow.utils.multimodal

Submodules

• lmflow.args
• lmflow.version

Package Contents

lmflow.internal_version = '0.0.7'

lmflow.__version__