diff --git a/docker/AI/docker-compose.yml b/docker/AI/docker-compose.yml
index f27e8be..0017015 100644
--- a/docker/AI/docker-compose.yml
+++ b/docker/AI/docker-compose.yml
@@ -1,6 +1,6 @@
 services:
   ollama:
-    image: ollama/ollama:0.5.1
+    image: ollama/ollama:0.5.4
     container_name: ollama
     restart: unless-stopped
     networks: