Skip to content

Commit e7868d3

Browse files
committed
add comments
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent a6cefaf commit e7868d3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

optimum/intel/ipex/modeling_base.py

+1
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@
6161
_IPEX_SUPPORT_MODEL_TYPES = ("llama", "bert", "vit", "falcon", "gpt2", "qwen2")
6262
_IPEX_EXPORTED_GENERATION_METHODS = ("sample", "greedy_search", "beam_sample", "beam_search", "assisted_generation")
6363
_IPEX_MINIMUM_VERSION_FOR_COMPILE = "2.5.0"
64+
# Page attention model cannot use torch.compile for now.
6465
_COMPILE_NOT_READY_MODEL_TYPES = ("llama", "falcon", "gpt2", "qwen2")
6566

6667

0 commit comments

Comments
 (0)