refine

intel-analytics · Mar 25, 2024 · d7e9700 · d7e9700
1 parent 7b8c7ae
commit d7e9700
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/llm/src/ipex_llm/transformers/speculative.py b/python/llm/src/ipex_llm/transformers/speculative.py
@@ -57,7 +57,7 @@ def generate(
         _enable_ipex = get_enable_ipex()
         if _enable_ipex and inputs.size(1) < 256:
             logger.warning(
-                "IPEX optimized models have issues for speculative decoding with short prompts"
+                "IPEX_CPU optimized models have issues for speculative decoding with short prompts"
                 "(length < 256). Using normal generate() method instead."
             )
             for var in ['max_step_draft', 'th_stop_draft', 'hf_adjust',