xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-23 12:31:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Minor

Browse Source
This commit is contained in:
Woosuk Kwon 2024-04-16 17:04:46 +00:00
parent b3b89cf755
commit de82e95787
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/executor/tpu_executor.py
View File

@ -27,6 +27,8 @@ class TPUExecutor(ExecutorBase):
self.parallel_config, self.parallel_config,
self.scheduler_config, self.scheduler_config,
self.device_config, self.device_config,
self.cache_config,
self.vision_language_config,
) )
self.driver_worker.init_device() self.driver_worker.init_device()
self.driver_worker.load_model() self.driver_worker.load_model()