Typo in mlx tensor support (#29509)

Potential typo in mlx support

Typo in mlx tensor support (#29509)
Potential typo in mlx support
631fa7bf · Pedro Cuenca · GitHub · b338a6c3 · 631fa7bf
Unverified Commit 631fa7bf authored Mar 08, 2024 by Pedro Cuenca Committed by GitHub Mar 08, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/utils/generic.py src/transformers/utils/generic.py +1 -1

No files found.
--- a/src/transformers/utils/generic.py
+++ b/src/transformers/utils/generic.py
@@ -242,7 +242,7 @@ def is_jax_tensor(x):


 def _is_mlx(x):
-    import mx.core as mx
+    import mlx.core as mx

    return isinstance(x, mx.array)