From 48a827aa07894e9734a72f394cced9ae0e6ef79d Mon Sep 17 00:00:00 2001 From: "Huang, Xinshengzi" Date: Thu, 22 Aug 2024 11:35:47 +0800 Subject: [PATCH] fix typos --- python/llm/src/ipex_llm/transformers/models/baichuan.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/llm/src/ipex_llm/transformers/models/baichuan.py b/python/llm/src/ipex_llm/transformers/models/baichuan.py index d454a97c..84416832 100644 --- a/python/llm/src/ipex_llm/transformers/models/baichuan.py +++ b/python/llm/src/ipex_llm/transformers/models/baichuan.py @@ -39,6 +39,7 @@ import os KV_CACHE_ALLOC_BLOCK_LENGTH = int(os.environ.get("KV_CACHE_ALLOC_BLOCK_LENGTH", 256)) + def pre_compute_inv_freq(module: torch.nn.Module): if module.__class__.__name__ == "RotaryEmbedding": inv_freq = module.inv_freq