Update modeling_quiet.py

modeling_quiet.py

@@ -1325,7 +1325,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
             nn.init.xavier_uniform_(module.weight)
             
     @torch.no_grad()
-    def generate(self, input_ids, attention_mask=None, **kwargs):
+    def generate(self, input_ids, attention_mask=None, streamer=None, **kwargs):
         if attention_mask is None:
             attention_mask = torch.ones_like(input_ids)