Skip to content

Commit 0de3ce2

Browse files
Fix problem with device request not being list
1 parent f6edb4a commit 0de3ce2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

inference_cli/lib/container_adapter.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,9 +84,9 @@ def start_inference_container(
8484
privileged = False
8585
if "gpu" in image:
8686
privileged = True
87-
device_requests = (
88-
[docker.types.DeviceRequest(device_ids=["all"], capabilities=[["gpu"]])],
89-
)
87+
device_requests = [
88+
docker.types.DeviceRequest(device_ids=["all"], capabilities=[["gpu"]])
89+
]
9090

9191
print(f"Starting inference server container...")
9292
docker_client.containers.run(

0 commit comments

Comments
 (0)