Update model.py

model.py

@@ -1054,11 +1054,13 @@ class Echo(nn.Module):
         for name, module in self.encoder.named_modules():
             if isinstance(module, (rotary)):
                 module.return_f0(f0)
+                module.update_base(f0)
 
         for name, module in self.decoder.named_modules():
             if isinstance(module, (rotary)):
                 module.return_f0(f0)
-   
+                module.update_base(f0)
+                
     def set_alignment_head(self, dump: bytes):
         array = np.frombuffer(
             gzip.decompress(base64.b85decode(dump)), dtype=bool).copy()