We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2f9139c commit 9497a45Copy full SHA for 9497a45
modules/ollama/testcontainers/ollama/__init__.py
@@ -101,7 +101,7 @@ def __init__(
101
def _check_and_add_gpu_capabilities(self):
102
info = self.get_docker_client().client.info()
103
if "nvidia" in info["Runtimes"]:
104
- self._kwargs = {**self._kwargs, "device_requests": DeviceRequest(count=-1, capabilities=[["gpu"]])}
+ self._kwargs = {**self._kwargs, "device_requests": [DeviceRequest(count=-1, capabilities=[["gpu"]])]}
105
106
def start(self) -> "OllamaContainer":
107
"""
0 commit comments