Skip to content

Commit

Permalink
Add assertions in JambaForCausalLM about not supporting chunked prefi…
Browse files Browse the repository at this point in the history
…ll and prefix caching
  • Loading branch information
tomeras91 committed Jul 25, 2024
1 parent 316a41a commit 9301493
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions vllm/model_executor/models/jamba.py
Original file line number Diff line number Diff line change
Expand Up @@ -644,6 +644,11 @@ def __init__(
lora_config: Optional[LoRAConfig] = None,
scheduler_config: Optional[SchedulerConfig] = None,
) -> None:
assert not scheduler_config.chunked_prefill_enabled, \
"Jamba currently does not support chunked prefill"
assert not cache_config.enable_prefix_caching, \
"Jamba currently does not support prefix caching"

super().__init__()
self.config = config
self.scheduler_config = scheduler_config
Expand Down

0 comments on commit 9301493

Please sign in to comment.