Skip to content

GitLab

"test/git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "96766101b4d181f9b3141da9484ca10ff7656743"
Commit 2cff4bd8 authored by Lysandre's avatar Lysandre Committed by Lysandre Debut
Browse files

Fix segmentation fault

parent f0616062
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 8 deletions
transformers/file_utils.py
View file @ 2cff4bd8
...@@ -26,14 +26,6 @@ from contextlib import contextmanager ...@@ -26,14 +26,6 @@ from contextlib import contextmanager
logger = logging.getLogger(__name__) # pylint: disable=invalid-name logger = logging.getLogger(__name__) # pylint: disable=invalid-name
try:
import tensorflow as tf
assert hasattr(tf, '__version__') and int(tf.__version__[0]) >= 2
_tf_available = True # pylint: disable=invalid-name
logger.info("TensorFlow version {} available.".format(tf.__version__))
except (ImportError, AssertionError):
_tf_available = False # pylint: disable=invalid-name
try: try:
import torch import torch
_torch_available = True # pylint: disable=invalid-name _torch_available = True # pylint: disable=invalid-name
...@@ -41,6 +33,13 @@ try: ...@@ -41,6 +33,13 @@ try:
except ImportError: except ImportError:
_torch_available = False # pylint: disable=invalid-name _torch_available = False # pylint: disable=invalid-name
try:
import tensorflow as tf
assert hasattr(tf, '__version__') and int(tf.__version__[0]) >= 2
_tf_available = True # pylint: disable=invalid-name
logger.info("TensorFlow version {} available.".format(tf.__version__))
except (ImportError, AssertionError):
_tf_available = False # pylint: disable=invalid-name
try: try:
from torch.hub import _get_torch_home from torch.hub import _get_torch_home
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment