Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fixed merge conflict
Browse files Browse the repository at this point in the history
Signed-off-by: Akshat Tripathi <akshat@krai.ai>
Akshat-Tripathi committed Dec 12, 2024
1 parent 49f6b12 commit be461b2
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion vllm/lora/punica_wrapper/punica_selector.py
Original file line number Diff line number Diff line change
@@ -15,5 +15,10 @@ def get_punica_wrapper(*args, **kwargs) -> PunicaWrapperBase:
from vllm.lora.punica_wrapper.punica_cpu import PunicaWrapperCPU
print_info_once("Using PunicaWrapperCPU.")
return PunicaWrapperCPU(*args, **kwargs)
elif current_platform.is_hpu():
# Lazy import to avoid ImportError
from vllm.lora.punica_wrapper.punica_hpu import PunicaWrapperHPU
print_info_once("Using PunicaWrapperHPU.")
return PunicaWrapperHPU(*args, **kwargs)
else:
raise NotImplementedError
raise NotImplementedError

0 comments on commit be461b2

Please sign in to comment.