[partition] relax tp and pp validations in partition script, use defa…

…ults in line with JIT UX
deepjavalibrary · Feb 4, 2025 · 3da4548 · 3da4548
1 parent 4ffed1f
commit 3da4548
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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()