Crystalcareai
commited on
Update modeling_quiet.py
Browse files- 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
|