orionweller
/

test-flex-gpt

Model card Files Files and versions Community

oweller2 commited on Nov 21

Commit

d4cfde8

•

1 Parent(s): 306d8ae

add breakpoint

Files changed (1) hide show

padding.py +1 -0

padding.py CHANGED Viewed

@@ -73,6 +73,7 @@ def pad_input(
         output[indices] = inputs
         padded_inputs = output.view(batch, seqlen)
     else:
         _, *rest = inputs.shape
         output = torch.zeros(batch * seqlen, *rest, dtype=inputs.dtype, device=inputs.device)
         output[indices] = inputs

         output[indices] = inputs
         padded_inputs = output.view(batch, seqlen)
     else:
+        breakpoint()
         _, *rest = inputs.shape
         output = torch.zeros(batch * seqlen, *rest, dtype=inputs.dtype, device=inputs.device)
         output[indices] = inputs