For testing

Author: krishung5

src/model.py

@@ -207,6 +207,12 @@ 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

@@ -49,4 +49,3 @@ conda pack -o pb_exec_env_model.py.tar.gz
 
 # deactivate conda
 conda deactivate
-conda deactivate