For testing

triton-inference-server · Nov 14, 2024 · 4487087 · 4487087
1 parent 63598a2
commit 4487087
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/model.py b/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__()
 

diff --git a/tools/gen_pb_exec_env.sh b/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
Original file line number	Diff line number	Diff line change
Expand Up		@@ -49,4 +49,3 @@ conda pack -o pb_exec_env_model.py.tar.gz

		# deactivate conda
		conda deactivate
		conda deactivate