Fix erroneous "model doesn't support compile" warning (#16486)
Signed-off-by: rzou <zou3519@gmail.com>
This commit is contained in:
parent
34b2cf3b33
commit
70de35a881
@ -3880,7 +3880,9 @@ def set_current_vllm_config(vllm_config: VllmConfig, check_compile=False):
|
|||||||
try:
|
try:
|
||||||
_current_vllm_config = vllm_config
|
_current_vllm_config = vllm_config
|
||||||
yield
|
yield
|
||||||
finally:
|
except Exception:
|
||||||
|
raise
|
||||||
|
else:
|
||||||
logger.debug("enabled custom ops: %s",
|
logger.debug("enabled custom ops: %s",
|
||||||
vllm_config.compilation_config.enabled_custom_ops)
|
vllm_config.compilation_config.enabled_custom_ops)
|
||||||
logger.debug("disabled custom ops: %s",
|
logger.debug("disabled custom ops: %s",
|
||||||
@ -3898,6 +3900,7 @@ def set_current_vllm_config(vllm_config: VllmConfig, check_compile=False):
|
|||||||
" does not support it. Please open an issue on GitHub"
|
" does not support it. Please open an issue on GitHub"
|
||||||
" if you want it to be supported.",
|
" if you want it to be supported.",
|
||||||
vllm_config.model_config.model)
|
vllm_config.model_config.model)
|
||||||
|
finally:
|
||||||
_current_vllm_config = old_vllm_config
|
_current_vllm_config = old_vllm_config
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user