From 46bdbf2576d04c90b912f0ad479bf2ee59bca3ac Mon Sep 17 00:00:00 2001 From: Justus Calvin Date: Tue, 12 Sep 2023 17:50:12 -0700 Subject: [PATCH] Remove duplicate code. --- src/together/finetune.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/src/together/finetune.py b/src/together/finetune.py index 1646b0d5..28680770 100644 --- 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"]