ai-models/Index-1.9B-32K
2
0
Fork 0
mirror of https://www.modelscope.cn/IndexTeam/Index-1.9B-32K.git synced 2026-04-02 13:52:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README_zh.md

Browse Source
This commit is contained in:
Cherrytest
2025-03-28 06:57:43 +00:00
parent 681c0afdb4
commit 6b19c1524e
6 changed files with 116 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modeling_index.py
View File

@ -111,8 +111,8 @@ class IndexRotaryEmbedding(torch.nn.Module):
self._set_cos_sin_cache(
seq_len=self.max_position_embeddings, device=self.inv_freq.device, dtype=torch.get_default_dtype()
)
print("rope cache value:")
self._show_rope_cache_info()
#print("rope cache value:")
#self._show_rope_cache_info()
def _show_rope_cache_info( self ):
np.set_printoptions(precision=6, threshold=10, edgeitems=5, linewidth=200, suppress=True)