Update modeling_quiet.py

modeling_quiet.py

@@ -1880,8 +1880,6 @@ class QuietForCausalLM(QuietPreTrainedModel):
                         attention_mask = attention_mask.expand(batch_size, 1, seq_len, seq_len)
                     elif attention_mask.dim() != 4:
                         raise ValueError(f"Attention mask should be of shape (batch_size, 1, seq_len, seq_len), but got {attention_mask.shape}")
-    
-    attention_mask = attention_mask.to(dtype=torch.bool, device=input_ids.device)
                 past_key_values = outputs.past_key_values
                 position_ids = position_ids + 1