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"]])
]
raise Exception(
f"""
Docoker image: {DOCKER_IMAGE}
args: {args}
container name: {container_name}
env: {env}
device_requests: {device_requests}
devices: {devices}
"""
)
# raise Exception(
# f"""
# Docoker image: {DOCKER_IMAGE}
# args: {args}
# container name: {container_name}
# env: {env}
# device_requests: {device_requests}
# devices: {devices}
# """
# )
env.pop("LOG_LEVEL")
env.pop("ROCR_VISIBLE_DEVICES")
container = client.containers.run(
DOCKER_IMAGE,
command=args,