From 8db03f3ab28b2ae98173352dafc189d0b7cd9837 Mon Sep 17 00:00:00 2001
From: Justin Hayes <justinh@rahb.ca>
Date: Fri, 5 Apr 2024 23:46:20 -0400
Subject: [PATCH] Add variables

---
 docker-compose.amdgpu.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker-compose.amdgpu.yaml b/docker-compose.amdgpu.yaml
index 130b1ccb7..7a1295d94 100644
--- a/docker-compose.amdgpu.yaml
+++ b/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}'
\ No newline at end of file
+      - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION-11.0.0}'
\ No newline at end of file