mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-01 18:51:14 +06:00
parent
b338a6c3b8
commit
631fa7bf6b
@ -242,7 +242,7 @@ def is_jax_tensor(x):
|
|||||||
|
|
||||||
|
|
||||||
def _is_mlx(x):
|
def _is_mlx(x):
|
||||||
import mx.core as mx
|
import mlx.core as mx
|
||||||
|
|
||||||
return isinstance(x, mx.array)
|
return isinstance(x, mx.array)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user