Update modeling_auristream.py

modeling_auristream.py

@@ -199,7 +199,7 @@ class AuriStream(PreTrainedModel):
         return logits, None
 
     def sample_logits(self, logits: torch.FloatTensor, temperature: float = 0.9, 
-                      top_k: int = 500, top_p: float = 0.5) -> torch.LongTensor:
+                      top_k: int = None, top_p: float = None) -> torch.LongTensor:
         """
         Samples an integer from the distribution of logits
         Parameters:
@@ -251,7 +251,7 @@ class AuriStream(PreTrainedModel):
 
     @torch.no_grad()
     def generate(self, seq: torch.Tensor, n_tokens: int = 1, temp=1.0, 
-        top_k=500, top_p=0.5, seed=None):
+        top_k=None, top_p=None, seed=None):
         """
         Parameters: 
             seq: torch.Tensor of shape (b, t, n_freq_bins)