From e94c3c71d175bd94bf0bd9291fdbd7fd9e814ff2 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 25 Dec 2024 21:36:22 +0530 Subject: [PATCH] revert configs. --- accelerate_configs/deepspeed.yaml | 4 ++-- accelerate_configs/uncompiled_2.yaml | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/accelerate_configs/deepspeed.yaml b/accelerate_configs/deepspeed.yaml index efbbf6f..62db0b4 100644 --- a/accelerate_configs/deepspeed.yaml +++ b/accelerate_configs/deepspeed.yaml @@ -14,10 +14,10 @@ machine_rank: 0 main_training_function: main mixed_precision: bf16 num_machines: 1 -num_processes: 1 +num_processes: 2 rdzv_backend: static same_network: true tpu_env: [] tpu_use_cluster: false tpu_use_sudo: false -use_cpu: false +use_cpu: false \ No newline at end of file diff --git a/accelerate_configs/uncompiled_2.yaml b/accelerate_configs/uncompiled_2.yaml index 0c99039..830b6e0 100644 --- a/accelerate_configs/uncompiled_2.yaml +++ b/accelerate_configs/uncompiled_2.yaml @@ -1,17 +1,17 @@ compute_environment: LOCAL_MACHINE debug: false -distributed_type: 'NO' +distributed_type: MULTI_GPU downcast_bf16: 'no' enable_cpu_affinity: false -gpu_ids: 0 +gpu_ids: 0,1 machine_rank: 0 main_training_function: main mixed_precision: bf16 num_machines: 1 -num_processes: 1 +num_processes: 2 rdzv_backend: static same_network: true tpu_env: [] tpu_use_cluster: false tpu_use_sudo: false -use_cpu: false +use_cpu: false \ No newline at end of file