remove changes

vwxyzjn · Aug 18, 2024 · 9ec06f2 · 9ec06f2
1 parent ca748a4
commit 9ec06f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/worker/cache_engine.py b/vllm/worker/cache_engine.py
@@ -64,7 +64,7 @@ def __init__(
 
         # Initialize the cache.
         self.gpu_cache = self._allocate_kv_cache(
-            self.num_gpu_blocks, self.device_config.device)
+            self.num_gpu_blocks, self.device_config.device_type)
         self.cpu_cache = self._allocate_kv_cache(self.num_cpu_blocks, "cpu")
 
     def _allocate_kv_cache(