riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

fix type check (#12638)

Browse Source
This commit is contained in:
Suraj Patil 2021-07-12 15:18:43 +05:30 committed by GitHub
parent 2dd9440d08
commit f8f9a679a0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/file_utils.py
View File

@ -1715,10 +1715,10 @@ def is_tensor(x):
return True
if is_flax_available():
import jaxlib.xla_extension as jax_xla
import jax.numpy as jnp
from jax.core import Tracer
if isinstance(x, (jax_xla.DeviceArray, Tracer)):
if isinstance(x, (jnp.ndarray, Tracer)):
return True
return isinstance(x, np.ndarray)