togethercomputer
/

StripedHyena-Nous-7B

Text Generation

Model card Files Files and versions Community

Zymrael commited on Dec 8, 2023

Commit

baa0d4e

•

1 Parent(s): 2ab6c7b

chore: small fix in dec

Files changed (1) hide show

engine.py +1 -1

engine.py CHANGED Viewed

@@ -230,7 +230,7 @@ class HyenaInferenceEngine:
         if iir_groups > 1:
             raise NotImplementedError
-        y = x2 * res_state + D * x1v
         return y, iir_state

         if iir_groups > 1:
             raise NotImplementedError
+        y = x2 * (res_state + D * x1v)
         return y, iir_state