replicated_linear.py 516 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project

from __future__ import annotations

from vllm.lora.layers.replicated_linear import ReplicatedLinearWithLoRA

from .base_linear import DiffusionBaseLinearLayerWithLoRA


class DiffusionReplicatedLinearWithLoRA(
    DiffusionBaseLinearLayerWithLoRA,
    ReplicatedLinearWithLoRA,
):
    """
    Diffusion ReplicatedLinear with LoRA.
    Prioritize apply() in DiffusionBaseLinearLayerWithLoRA
    """

    pass