diff --git a/ollama/docker-compose.yml b/ollama/docker-compose.yml
index ac2138d..74c3f6c 100755
--- a/ollama/docker-compose.yml
+++ b/ollama/docker-compose.yml
@@ -1,6 +1,6 @@
 services:
   ollama:
-    image: ${DOCKER_REGISTRY:-}ollama/ollama:rocm
+    image: ${DOCKER_REGISTRY:-}ollama/ollama
     container_name: ollama
     restart: ${RESTART:-unless-stopped}
     devices:
@@ -10,6 +10,7 @@ services:
       - models:/root/.ollama
     environment:
       - HSA_OVERRIDE_GFX_VERSION=10.3.0
+      - OLLAMA_VULKAN=1 
     ports:
       - "11434:11434"
     dns: