Skip to content

Commit

Permalink
Remove dead code
Browse files Browse the repository at this point in the history
  • Loading branch information
xenova committed Apr 6, 2024
1 parent ee53edc commit 9911186
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions src/models.js
Original file line number Diff line number Diff line change
Expand Up @@ -1065,13 +1065,6 @@ export class PreTrainedModel extends Callable {
const decoder_input_ids = decoder_input_ids_start;
model_kwargs['decoder_attention_mask'] = ones_like(decoder_input_ids);

// if (generation_config.guidance_scale !== null && generation_config.guidance_scale > 1) {
// model_kwargs['decoder_attention_mask'] = cat([
// model_kwargs['decoder_attention_mask'],
// zeros_like(model_kwargs['decoder_attention_mask']),
// ], 0)
// }

return { input_ids: decoder_input_ids, model_inputs: model_kwargs };
}

Expand Down

0 comments on commit 9911186

Please sign in to comment.