diff --git a/src/model.py b/src/model.py index 5a251e5..d8ed413 100755 --- a/src/model.py +++ b/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__() diff --git a/tools/gen_pb_exec_env.sh b/tools/gen_pb_exec_env.sh index d9faf3a..d01a0d9 100755 --- a/tools/gen_pb_exec_env.sh +++ b/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