fix

2026-02-24 12:10:09 +01:00 · 2023-08-01 10:15:01 +08:00
parent 34b75afac3
commit f047eed4c1
1 changed files with 1 additions and 0 deletions
--- a/modelscope/models/nlp/chatglm2/text_generation.py
+++ b/modelscope/models/nlp/chatglm2/text_generation.py
@@ -134,6 +134,7 @@ class RotaryEmbedding(nn.Module):
        self.register_buffer('inv_freq', inv_freq)
        self.dim = dim
        self.original_impl = original_impl
+        self.rope_ratio = rope_ratio

    def forward_impl(self,
                     seq_len: int,