espnet2.speechlm.model.speechlm.lm.parallel_pp.ParallelPPHFModel

source

espnet2.speechlm.model.speechlm.lm.parallel_pp.ParallelPPHFModel(model_hf_tag, **kwargs)

Factory function to create a PP-aware parallel multimodal LLM.

• Parameters:
  • model_hf_tag – HuggingFace model identifier
  • **kwargs – Must include parallel_dims and pp_layout. Remaining kwargs are forwarded to from_pretrained.
• Returns: Instantiated stage-local model with only this rank’s modules.