From 3da4548de52eb0ec2df6bb5da1ec3ee45e34304f Mon Sep 17 00:00:00 2001 From: Siddharth Venkatesan Date: Mon, 3 Feb 2025 12:45:07 -0800 Subject: [PATCH] [partition] relax tp and pp validations in partition script, use defaults in line with JIT UX --- serving/docker/partition/properties_manager.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/serving/docker/partition/properties_manager.py b/serving/docker/partition/properties_manager.py index eff9bf4f1..35c1316b6 100644 --- a/serving/docker/partition/properties_manager.py +++ b/serving/docker/partition/properties_manager.py @@ -57,10 +57,6 @@ def __init__(self, args, **kwargs): self.set_and_validate_model_dir() - if self.is_mpi_mode: - self.validate_tp_degree() - self.validate_pp_degree() - self.set_and_validate_entry_point() self.set_and_validate_save_mp_checkpoint_path()