diff --git a/docker-compose.yml b/docker-compose.yml
index 06add03..db972cb 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -6,6 +6,7 @@ services:
     environment:
       - HF_TOKEN=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
       - PYTORCH_ALLOC_CONF=expandable_segments:True
+      - VLLM_DEVICE=cpu
     ipc: host
     ports:
       - "8000:8000"