initial commit (#1174)

neuralmagic · Aug 8, 2023 · 91f6774 · 91f6774
1 parent a0b6088
commit 91f6774
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/deepsparse/transformers/utils/helpers.py b/src/deepsparse/transformers/utils/helpers.py
@@ -61,7 +61,7 @@ def create_causal_mask(
     dtype: numpy.dtype = numpy.int64,
 ) -> numpy.ndarray:
     """
-    Compute a causal from a set of module inputs.
+    Compute a causal mask from a set of module inputs.
     In transformers, a causal mask is a boolean mask that is used to
     prevent information from future positions in a sequence from
     being used to predict the current position. Each element of the mask