Fix minor bug in max_mem_gb_classifier scripts.py

Signed-off-by: Vibhu Jawa <[email protected]>

nemo_curator/utils/script_utils.py

@@ -450,6 +450,7 @@ def add_distributed_classifier_cluster_args(self):
         self.add_distributed_args()
         self.add_arg_enable_spilling()
         self.add_arg_set_torch_to_use_rmm()
+        self.add_arg_max_mem_gb_classifier()
 
         # Set low default RMM pool size for classifier
         # to allow pytorch to grow its memory usage