Whoops

pytorch · Dec 6, 2024 · 1ad2910 · 1ad2910
1 parent 6030d9f
commit 1ad2910
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recipes/knowledge_distillation_distributed.py b/recipes/knowledge_distillation_distributed.py
@@ -254,7 +254,7 @@ def setup(self, cfg: DictConfig) -> None:
         )
 
         self._tokenizer = config.instantiate(cfg.tokenizer)
-        utils.log_rank_zero("Tokenizer is initialized from file.")
+        utils.log_rank_zero(log, "Tokenizer is initialized from file.")
 
         self._optimizer = self._setup_optimizer(
             cfg_optimizer=cfg.optimizer,