jinaai
/

xlm-roberta-flash-implementation

Inference Endpoints

🇪🇺 Region: EU

Model card Files Files and versions Community

jupyterjazz commited on Aug 23

Commit

169b7fb

•

1 Parent(s): 80b1ff7

Fix rotary backward

Files changed (1) hide show

rotary.py +1 -1

rotary.py CHANGED Viewed

@@ -290,7 +290,7 @@ class ApplyRotaryEmbQKV_(torch.autograd.Function):
                 cu_seqlens=cu_seqlens,
                 max_seqlen=ctx.max_seqlen,
             )
-        return dqkv, None, None, None, None, None, None, None, None
 def apply_rotary_emb_qkv_(

                 cu_seqlens=cu_seqlens,
                 max_seqlen=ctx.max_seqlen,
             )
+        return dqkv, None, None, None, None, None, None, None, None, None
 def apply_rotary_emb_qkv_(