fix: exit early if no adapter_data

This commit is contained in:
drbh 2024-06-25 16:20:33 +00:00
parent 0d496baaa4
commit a2d821c4a3

View File

@ -41,6 +41,8 @@ class LoraLinear(nn.Module):
start_idx: int,
end_idx: int,
) -> torch.Tensor:
if adapter_data is None:
return result
data = adapter_data.data.get(layer_type)
data: Optional["BatchLoraWeights"] = (
data.get("lora") if data is not None else None