diff --git a/servers/unraid/hindsight-retrieval/compose.yaml b/servers/unraid/hindsight-retrieval/compose.yaml index 560f549..aee3241 100644 --- a/servers/unraid/hindsight-retrieval/compose.yaml +++ b/servers/unraid/hindsight-retrieval/compose.yaml @@ -11,6 +11,7 @@ services: - float16 environment: HF_HUB_CACHE: /data + NVIDIA_VISIBLE_DEVICES: "0" HTTP_PROXY: http://192.168.50.45:7890 HTTPS_PROXY: http://192.168.50.45:7890 NO_PROXY: 127.0.0.1,localhost,192.168.50.0/24 @@ -49,6 +50,7 @@ services: - float16 environment: HF_HUB_CACHE: /data + NVIDIA_VISIBLE_DEVICES: "0" HTTP_PROXY: http://192.168.50.45:7890 HTTPS_PROXY: http://192.168.50.45:7890 NO_PROXY: 127.0.0.1,localhost,192.168.50.0/24