hot-fix silu error import (#10292)

intel-analytics · Mar 1, 2024 · a1cebf3 · a1cebf3
1 parent 4df6dd8
commit a1cebf3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/llm/src/bigdl/llm/transformers/models/yuan.py b/python/llm/src/bigdl/llm/transformers/models/yuan.py
@@ -34,7 +34,7 @@
 from bigdl.llm.transformers.models.utils import init_kv_cache, extend_kv_cache, append_kv_cache
 from bigdl.llm.transformers.models.utils import init_fp8_kv_cache, append_fp8_kv_cache, \
     restore_fp8_kv_cache, use_quantize_kv_cache
-from bigdl.llm.transformers.models.utils import is_enough_kv_cache_room_4_31. SILU
+from bigdl.llm.transformers.models.utils import is_enough_kv_cache_room_4_31, SILU
 from bigdl.llm.transformers.low_bit_linear import SYM_INT4, FP8E5
 
 KV_CACHE_ALLOC_BLOCK_LENGTH = 256