Remove duplicate code.

togethercomputer · Sep 13, 2023 · 46bdbf2 · 46bdbf2
1 parent 499df3e
commit 46bdbf2
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/src/together/finetune.py b/src/together/finetune.py
@@ -89,16 +89,6 @@ def create(
                 f"The number of checkpoints must be < the number of epochs, setting to {n_checkpoints}"
             )
 
-        # Set default batch size
-        if batch_size is None:
-            if model in [
-                "togethercomputer/llama-2-70b",
-                "togethercomputer/llama-2-70b-chat",
-            ]:
-                batch_size = 144
-            else:
-                batch_size = 32
-
         if (
             model
             in ["togethercomputer/llama-2-70b", "togethercomputer/llama-2-70b-chat"]