From 0936c79995c6acd9ca25d0a2786577d76c82d57a Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Wed, 2 Oct 2024 18:07:13 +0200 Subject: [PATCH] Raw tgi --- integration-tests/conftest.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index 2808b950..eaea5911 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -527,21 +527,21 @@ def launcher(event_loop): # devices: {devices} # """ # ) - env.pop("LOG_LEVEL") - env.pop("ROCR_VISIBLE_DEVICES") + # env.pop("LOG_LEVEL") + # env.pop("ROCR_VISIBLE_DEVICES") container = client.containers.run( DOCKER_IMAGE, - command=args, + # command=args, # name=container_name, - environment=env, - auto_remove=False, - detach=True, - device_requests=device_requests, - devices=devices, - volumes=volumes, - ports={"80/tcp": port}, - healthcheck={"timeout": int(10 * 1e9)}, - shm_size="1G", + # environment=env, + # auto_remove=False, + # detach=True, + # device_requests=device_requests, + # devices=devices, + # volumes=volumes, + # ports={"80/tcp": port}, + # healthcheck={"timeout": int(10 * 1e9)}, + # shm_size="1G", ) import time