We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e7868d3 commit 7becc20Copy full SHA for 7becc20
optimum/intel/ipex/modeling_base.py
@@ -61,7 +61,7 @@
61
_IPEX_SUPPORT_MODEL_TYPES = ("llama", "bert", "vit", "falcon", "gpt2", "qwen2")
62
_IPEX_EXPORTED_GENERATION_METHODS = ("sample", "greedy_search", "beam_sample", "beam_search", "assisted_generation")
63
_IPEX_MINIMUM_VERSION_FOR_COMPILE = "2.5.0"
64
-# Page attention model cannot use torch.compile for now.
+# Page attention model cannot use torch.compile for now.
65
_COMPILE_NOT_READY_MODEL_TYPES = ("llama", "falcon", "gpt2", "qwen2")
66
67
0 commit comments