Try with python 3.12

Author: krishung5

src/model.py

@@ -207,12 +207,6 @@ def initialize(self, args):
             "Torch parallelism settings " + for_model + ": " + setting_msg
         )
 
-        # Print out python version
-        print(
-            f"Python version is {sys.version_info.major}.{sys.version_info.minor}",
-            flush=True,
-        )
-
         self._infer_mode = torch.inference_mode(mode=True)
         self._infer_mode.__enter__()
 

tools/gen_pb_exec_env.sh

@@ -32,7 +32,7 @@ bash Miniconda3-py312_24.9.2-0-Linux-x86_64.sh -p ./miniconda -b
 eval "$(./miniconda/bin/conda shell.bash hook)"
 
 # create conda environment
-conda create -n pt python=3.10 -y
+conda create -n pt python=3.12 -y
 conda activate pt
 conda install -c conda-forge conda-pack -y