Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
aa4afa67
Unverified
Commit
aa4afa67
authored
Jul 19, 2023
by
Madhava Jay
Committed by
GitHub
Jul 19, 2023
Browse files
Fixed issue where ACCELERATE_USE_CPU="False" results in bool(True) (#24907)
- This results in cpu mode on Apple Silicon mps
parent
243b2ea3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
src/transformers/training_args.py
src/transformers/training_args.py
+2
-1
No files found.
src/transformers/training_args.py
View file @
aa4afa67
...
...
@@ -53,6 +53,7 @@ from .utils import (
logging
,
requires_backends
,
)
from
.utils.generic
import
strtobool
from
.utils.import_utils
import
is_optimum_neuron_available
...
...
@@ -1720,7 +1721,7 @@ class TrainingArguments:
self
.
distributed_state
=
None
if
not
self
.
use_ipex
and
"ACCELERATE_USE_IPEX"
not
in
os
.
environ
:
os
.
environ
[
"ACCELERATE_USE_IPEX"
]
=
"false"
if
self
.
use_cpu
or
os
.
environ
.
get
(
"ACCELERATE_USE_CPU"
,
False
):
if
self
.
use_cpu
or
strtobool
(
os
.
environ
.
get
(
"ACCELERATE_USE_CPU"
,
"
False
"
)
):
self
.
distributed_state
=
PartialState
(
cpu
=
True
,
backend
=
self
.
ddp_backend
)
self
.
_n_gpu
=
0
elif
is_sagemaker_mp_enabled
():
...
...
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