Typo in mlx tensor support (#29509)

Potential typo in mlx support
huggingface · May 14, 2024 · 00ef7a0 · 00ef7a0
1 parent 20bc2bb
commit 00ef7a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)