Bläddra i källkod

Add variables

Justin Hayes 1 år sedan
förälder
incheckning
8db03f3ab2
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      docker-compose.amdgpu.yaml

+ 2 - 2
docker-compose.amdgpu.yaml

@@ -3,6 +3,6 @@ services:
     devices:
       - /dev/kfd:/dev/kfd
       - /dev/dri:/dev/dri
-    image: ollama/ollama:rocm
+    image: ollama/ollama:${OLLAMA_DOCKER_TAG-rocm}
     environment:
-      - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION}'
+      - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION-11.0.0}'