fix: update frequencies when updating the rope base value

rotary.py

@@ -495,6 +495,7 @@ class RotaryEmbedding(torch.nn.Module):
     def base(self, new_base):
         if new_base > 0:
             self._base = float(new_base)
+            self.inv_freq = self._compute_inv_freq(device=self.inv_freq.device)
         else:
             raise ValueError("Rotary base value must be positive")