diff --git a/torchchat/generate.py b/torchchat/generate.py
index 5ae7ecfad..826361cb8 100644
--- a/torchchat/generate.py
+++ b/torchchat/generate.py
@@ -1389,7 +1389,7 @@ def distributed_input(prompt: str) -> str:
         if builder_args.pp > 1:
             self.seqlen_prefill = 1024  # sequence length for prefill stage
 
-            logger.warn(
+            logger.warning(
                 f"{color.yellow}Pipeline parallelism is still experimental and might be slow{color.reset}"
             )
             pp_mesh = self.model.device_mesh["pp"]