diff --git a/finetune/finetune.py b/finetune/finetune.py index c752de085..b8e3747ff 100644 --- a/finetune/finetune.py +++ b/finetune/finetune.py @@ -31,7 +31,7 @@ def get_accelerate_environment_variable(mode: str, config: Dict[str, Any]) -> di mode_env_vars = { "CPU_DDP": { "ACCELERATE_USE_CPU": "true", - "ACCELERATE_USE_IPEX": "false", + "ACCELERATE_USE_IPEX": "true", "ACCELERATE_MIXED_PRECISION": mixed_precision, }, "GPU_DDP": {