Update model.py

model.py

@@ -349,7 +349,7 @@ class rotary(nn.Module):
     def update_base(self, pitch):
         theta = pitch.squeeze(0).to(device, dtype)
         f0_mean = theta.mean() + 1e-8
-        inv_freq = 1.0 / (f0_mean ** (torch.arange(start=0, end=self.dim, step=2, device=device, dtype=dtype) / self.dim))
+        inv_freq = 700 * (torch.pow(10, torch.linspace(0, 2595 * torch.log10(torch.tensor(1 + 8000/700)), dim // 2, device=device, dtype=dtype) / 2595) - 1) / 1000 
         self.inv_freq.data.copy_(inv_freq)
         self.theta.data.copy_(f0_mean)