diff --git a/modules/ollama/testcontainers/ollama/__init__.py b/modules/ollama/testcontainers/ollama/__init__.py index ea089f14..002b02d6 100644 --- a/modules/ollama/testcontainers/ollama/__init__.py +++ b/modules/ollama/testcontainers/ollama/__init__.py @@ -101,7 +101,7 @@ def __init__( def _check_and_add_gpu_capabilities(self): info = self.get_docker_client().client.info() if "nvidia" in info["Runtimes"]: - self._kwargs = {**self._kwargs, "device_requests": DeviceRequest(count=-1, capabilities=[["gpu"]])} + self._kwargs = {**self._kwargs, "device_requests": [DeviceRequest(count=-1, capabilities=[["gpu"]])]} def start(self) -> "OllamaContainer": """