From 62899bd85c7839939cd384caad53b2963a33a750 Mon Sep 17 00:00:00 2001
From: Misko <misko@meta.com>
Date: Tue, 6 Aug 2024 21:56:50 +0000
Subject: [PATCH] remove load balancing for painn tests

---
 tests/core/models/test_configs/test_painn_hydra.yml | 1 -
 1 file changed, 1 deletion(-)

diff --git a/tests/core/models/test_configs/test_painn_hydra.yml b/tests/core/models/test_configs/test_painn_hydra.yml
index 0b39aa173..09b3bb34b 100644
--- a/tests/core/models/test_configs/test_painn_hydra.yml
+++ b/tests/core/models/test_configs/test_painn_hydra.yml
@@ -38,7 +38,6 @@ model:
 optim:
   batch_size: 32
   eval_batch_size: 32
-  load_balancing: atoms
   eval_every: 5000
   num_workers: 2
   optimizer: AdamW