File tree 1 file changed +4
-2
lines changed
1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 44
44
)
45
45
from transformers .utils .generic import strtobool
46
46
47
+ from ..utils .import_utils import check_if_transformers_greater
48
+
47
49
48
50
if is_torch_available ():
49
51
import torch
50
52
51
- if is_accelerate_available ():
53
+ if is_accelerate_available () and check_if_transformers_greater ( "4.38.0" ) :
52
54
from transformers .trainer_pt_utils import AcceleratorConfig
53
55
54
56
@@ -449,7 +451,7 @@ def __post_init__(self):
449
451
os .environ [f"{ prefix } SYNC_MODULE_STATES" ] = self .fsdp_config .get ("sync_module_states" , "true" )
450
452
os .environ [f"{ prefix } USE_ORIG_PARAMS" ] = self .fsdp_config .get ("use_orig_params" , "false" )
451
453
452
- if is_accelerate_available ():
454
+ if is_accelerate_available () and check_if_transformers_greater ( "4.38.0" ) :
453
455
if not isinstance (self .accelerator_config , (AcceleratorConfig )):
454
456
if self .accelerator_config is None :
455
457
self .accelerator_config = AcceleratorConfig ()
You can’t perform that action at this time.
0 commit comments