Abdullah-Nazhat
/

MCDP_MAMBA

Model card Files Files and versions

Abdullah-Nazhat commited on Aug 25

Commit

910c8ce

·

verified ·

1 Parent(s): 65bd18d

Update pscan.py

Files changed (1) hide show

pscan.py +1 -1

pscan.py CHANGED Viewed

@@ -152,4 +152,4 @@ class PScan(torch.autograd.Function):
         return Q.transpose(2, 1)[:, :L], grad_output.transpose(2, 1)[:, :L]
-pscan = PScan.apply


152
153	return Q.transpose(2, 1)[:, :L], grad_output.transpose(2, 1)[:, :L]
154
155	+ pscan = PScan.apply