Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
fb2b8984
Unverified
Commit
fb2b8984
authored
Mar 23, 2021
by
Stas Bekman
Committed by
GitHub
Mar 23, 2021
Browse files
[file_utils] import refactor (#10859)
* import refactor * fix the fallback
parent
3f48b2bc
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
15 additions
and
22 deletions
+15
-22
src/transformers/file_utils.py
src/transformers/file_utils.py
+15
-22
No files found.
src/transformers/file_utils.py
View file @
fb2b8984
...
...
@@ -84,31 +84,24 @@ else:
if
USE_TF
in
ENV_VARS_TRUE_AND_AUTO_VALUES
and
USE_TORCH
not
in
ENV_VARS_TRUE_VALUES
:
_tf_available
=
importlib
.
util
.
find_spec
(
"tensorflow"
)
is
not
None
if
_tf_available
:
candidates
=
(
"tensorflow"
,
"tensorflow-cpu"
,
"tensorflow-gpu"
,
"tf-nightly"
,
"tf-nightly-cpu"
,
"tf-nightly-gpu"
,
"intel-tensorflow"
,
)
_tf_version
=
None
# For the metadata, we have to look for both tensorflow and tensorflow-cpu
try
:
_tf_version
=
importlib_metadata
.
version
(
"tensorflow"
)
except
importlib_metadata
.
PackageNotFoundError
:
for
pkg
in
candidates
:
try
:
_tf_version
=
importlib_metadata
.
version
(
"tensorflow-cpu"
)
_tf_version
=
importlib_metadata
.
version
(
pkg
)
break
except
importlib_metadata
.
PackageNotFoundError
:
try
:
_tf_version
=
importlib_metadata
.
version
(
"tensorflow-gpu"
)
except
importlib_metadata
.
PackageNotFoundError
:
try
:
_tf_version
=
importlib_metadata
.
version
(
"tf-nightly"
)
except
importlib_metadata
.
PackageNotFoundError
:
try
:
_tf_version
=
importlib_metadata
.
version
(
"tf-nightly-cpu"
)
except
importlib_metadata
.
PackageNotFoundError
:
try
:
_tf_version
=
importlib_metadata
.
version
(
"tf-nightly-gpu"
)
except
importlib_metadata
.
PackageNotFoundError
:
# Support for intel-tensorflow version
try
:
_tf_version
=
importlib_metadata
.
version
(
"intel-tensorflow"
)
except
importlib_metadata
.
PackageNotFoundError
:
_tf_version
=
None
_tf_available
=
False
pass
_tf_available
=
_tf_version
is
not
None
if
_tf_available
:
if
version
.
parse
(
_tf_version
)
<
version
.
parse
(
"2"
):
logger
.
info
(
f
"TensorFlow found but with version
{
_tf_version
}
. Transformers requires version 2 minimum."
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment