Skip to content

Commit 4ca67f5

Browse files
authored
fix lora (#4172)
1 parent 05a563e commit 4ca67f5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

lmdeploy/pytorch/nn/linear/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def _forward_default(self, x, all_reduce: bool, tp_sizes: List[int]):
181181
"""Default forward implement."""
182182
raise NotImplementedError('This method should be implemented in subclasses.')
183183

184-
def _forward_lora(self, x, tp_sizes: List[int]):
184+
def _forward_lora(self, x, tp_sizes: List[int] = None):
185185
"""Forward with LoRA."""
186186
out = self._forward_default(x, False, tp_sizes)
187187

0 commit comments

Comments
 (0)