Update src/levanter/checkpoint.py

Co-authored-by: David Hall <[email protected]>
stanford-crfm · Oct 28, 2024 · aa47d4e · aa47d4e
1 parent 7331774
commit aa47d4e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/levanter/checkpoint.py b/src/levanter/checkpoint.py
@@ -27,7 +27,6 @@
 
 from levanter.tensorstore_serialization import tree_deserialize_leaves_tensorstore, tree_serialize_leaves_tensorstore
 from levanter.types import FilterSpec
-# from levanter.trainer import StepInfo
 
 
 logger = logging.getLogger(__name__)