ModularPipeline
_tp_plan
_cp_plan
_workflow_map
device_map="balanced"
flash_varlen
flash_4_hub
return_lse
train_text_to_image_lora.py
train_lcm_distill_lora_sdxl.py