hotfix memory profiler none

PrimeIntellect-ai · Oct 2, 2024 · 0ba28db · 0ba28db
1 parent 2356b1c
commit 0ba28db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/zeroband/train.py b/src/zeroband/train.py
@@ -295,7 +295,7 @@ def train(config: Config):
 
             logger.info(log)
 
-            if memory_profiler is not None:
+            if config.train.memory_profiler is not None:
                 memory_profiler.step()
 
         if config.diloco is not None: