espnet2.speechlm.model.speechlm.lm.parallel_pp.build_parallel_pp_hf_class

source

espnet2.speechlm.model.speechlm.lm.parallel_pp.build_parallel_pp_hf_class(model_hf_tag)

Create a PP-aware LLM class that inherits from ParallelLLM.

The returned ParallelPPLLM overrides from_pretrained (to load only the local stage’s weights) and forward (to run stage-wise computation).