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

Renaming framework env variables flags from NO_ to USE_

Browse Source
This commit is contained in:
Morgan Funtowicz 2019-12-19 11:49:06 +01:00
parent f516cf3956
commit fc624716aa
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
transformers/file_utils.py
View File

@ -27,8 +27,9 @@ from contextlib import contextmanager
logger = logging.getLogger(__name__) # pylint: disable=invalid-name
try:
if 'NO_TF' in os.environ and os.environ['NO_TF'].upper() in ('1', 'ON'):
logger.info("Found NO_TF, disabling TensorFlow")
os.environ.setdefault('USE_TF', 'YES')
if os.environ['USE_TF'].upper() in ('1', 'ON', 'YES'):
logger.info("USE_TF override through env variable, disabling Tensorflow")
_tf_available = False
else:
import tensorflow as tf
@ -39,8 +40,9 @@ except (ImportError, AssertionError):
_tf_available = False # pylint: disable=invalid-name
try:
if 'NO_TORCH' in os.environ and os.environ['NO_TORCH'].upper() in ('1', 'ON'):
logger.info("Found NO_TORCH, disabling PyTorch")
os.environ.setdefault('USE_TORCH', 'YES')
if os.environ['USE_TORCH'].upper() in ('1', 'ON', 'YES'):
logger.info("USE_TORCH override through env variable, disabling PyTorch")
_torch_available = False
else:
import torch