Crystalcareai commited on
Commit
904dcda
·
verified ·
1 Parent(s): b52f8ef

Update modeling_quiet.py

Browse files
Files changed (1) hide show
  1. modeling_quiet.py +1 -1
modeling_quiet.py CHANGED
@@ -1371,7 +1371,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
1371
  next_token_id = torch.argmax(next_token_logits, dim=-1)
1372
 
1373
  # Append the generated token to the input sequence
1374
- input_ids = torch.cat([input_ids, next_token_id.unsqueeze(-1).to(input_ids.device)], dim=-1)
1375
  seq_len += 1
1376
 
1377
  # Update the attention mask
 
1371
  next_token_id = torch.argmax(next_token_logits, dim=-1)
1372
 
1373
  # Append the generated token to the input sequence
1374
+ # input_ids = torch.cat([input_ids, next_token_id.unsqueeze(-1).to(input_ids.device)], dim=-1)
1375
  seq_len += 1
1376
 
1377
  # Update the attention mask