override the entry point for vvllm

Author: Jyothirmaikottu

src/image.py

@@ -130,9 +130,15 @@ def collect_installed_packages_information(self):
         ]
         for command in commands:
             command_responses.append(f"\n{command}")
-            command_responses.append(
-                bytes.decode(docker_client.containers.run(self.ecr_url, command))
-            )
+            # Override entrypoint for vLLM images to prevent server startup
+            if "vllm" in self.ecr_url.lower():
+                command_responses.append(
+                    bytes.decode(docker_client.containers.run(self.ecr_url, command, entrypoint=""))
+                )
+            else:
+                command_responses.append(
+                    bytes.decode(docker_client.containers.run(self.ecr_url, command))
+                )
         docker_client.containers.prune()
         return command_responses