From 93338ce2462957f11f99eafb5fdb93637fe02776 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 20 Nov 2024 17:07:53 -0800 Subject: [PATCH] Update fast_lora.py --- unsloth/kernels/fast_lora.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/unsloth/kernels/fast_lora.py b/unsloth/kernels/fast_lora.py index 6481661d88..c2b7929a29 100644 --- a/unsloth/kernels/fast_lora.py +++ b/unsloth/kernels/fast_lora.py @@ -415,6 +415,9 @@ pass IDENTITY_DROPOUT = torch.nn.Identity @torch._disable_dynamo def fast_lora_forward(self, x: torch.Tensor, *args, **kwargs) -> torch.Tensor: + raise NotImplementedError( + "Unsloth: Currently not supported yet - reshaping done incorrectly" + ) self._check_forward_args(x, *args, **kwargs) adapter_names = kwargs.pop("adapter_names", None)