vllm_omni.diffusion.lora.utils ¶

from_layer_diffusion ¶

from_layer_diffusion(
    layer: Module,
    max_loras: int,
    lora_config: LoRAConfig,
    packed_modules_list: list[str],
    model_config: PretrainedConfig | None = None,
) -> Module

Diffusion-specific layer replacement. similar to vLLM's from_layer