diff --git a/immich/docker-compose.yml b/immich/docker-compose.yml
index aebfe09..9ca51e3 100644
--- a/immich/docker-compose.yml
+++ b/immich/docker-compose.yml
@@ -88,10 +88,12 @@ services:
     environment:
       - MACHINE_LEARNING_CACHE_FOLDER=/cache
       - MACHINE_LEARNING_REQUEST_TIMEOUT=600
+      - MACHINE_LEARNING_MODEL_TTL=300
+      - MACHINE_LEARNING_MODEL_TTL_POLL_S=10
     deploy:
       resources:
         limits:
-          memory: 4G
+          memory: 6G
     networks:
       - immich
     logging: