fix: remove double RMSNorm on B/C in MambaInnerFn.backward at checkpoint_lvl=1#939
Closed
Chessing234 wants to merge 1 commit into
Closed
fix: remove double RMSNorm on B/C in MambaInnerFn.backward at checkpoint_lvl=1#939Chessing234 wants to merge 1 commit into
Chessing234 wants to merge 1 commit into