From c16e94d740482f2d83424bf35de17714812e5d4f Mon Sep 17 00:00:00 2001
From: Daniel King <daniel@mosaicml.com>
Date: Sat, 22 Jun 2024 14:03:41 -0700
Subject: [PATCH] pc

---
 scripts/train/train.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/scripts/train/train.py b/scripts/train/train.py
index 5a05d18e94..4620254135 100644
--- a/scripts/train/train.py
+++ b/scripts/train/train.py
@@ -50,7 +50,6 @@
     make_dataclass_and_log_config,
     pop_config,
     process_init_device,
-    update_batch_size_info,
 )
 from llmfoundry.utils.exceptions import (
     BaseContextualError,