diff --git a/unsloth/kernels/dropout.py b/unsloth/kernels/dropout.py index 9d18e92d1c..184debefcf 100644 --- a/unsloth/kernels/dropout.py +++ b/unsloth/kernels/dropout.py @@ -63,6 +63,7 @@ def _seeded_dropout( tl.store(output_ptr + off2, o2, mask = mask2) tl.store(output_ptr + off3, o3, mask = mask3) +pass def seeded_dropout( x: torch.Tensor,