huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove stdin messing

Browse Source
This commit is contained in:
Nicolas Patry 2024-10-02 18:00:27 +02:00
parent 7bead9ef0f
commit 3ae4e090eb
No known key found for this signature in database
GPG Key ID: E939E8CC91A1C674
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
integration-tests/conftest.py
View File

@ -517,16 +517,18 @@ def launcher(event_loop):
docker.types.DeviceRequest(count=gpu_count, capabilities=[["gpu"]]) docker.types.DeviceRequest(count=gpu_count, capabilities=[["gpu"]])
] ]
raise Exception( # raise Exception(
f""" # f"""
Docoker image: {DOCKER_IMAGE} # Docoker image: {DOCKER_IMAGE}
args: {args} # args: {args}
container name: {container_name} # container name: {container_name}
env: {env} # env: {env}
device_requests: {device_requests} # device_requests: {device_requests}
devices: {devices} # devices: {devices}
""" # """
) # )
env.pop("LOG_LEVEL")
env.pop("ROCR_VISIBLE_DEVICES")
container = client.containers.run( container = client.containers.run(
DOCKER_IMAGE, DOCKER_IMAGE,
command=args, command=args,