Class: Informers::Utils::ForcedEOSTokenLogitsProcessor

Inherits:

LogitsProcessor

• Object
• LogitsProcessor
• Informers::Utils::ForcedEOSTokenLogitsProcessor

Defined in:

lib/informers/utils/generation.rb

Instance Method Summary

• #call(input_ids, logits) ⇒ Object
• #initialize(max_length, forced_eos_token_id) ⇒ ForcedEOSTokenLogitsProcessor constructor

  A new instance of ForcedEOSTokenLogitsProcessor.

Constructor Details

#initialize(max_length, forced_eos_token_id) ⇒ ForcedEOSTokenLogitsProcessor

Returns a new instance of ForcedEOSTokenLogitsProcessor.

# File 'lib/informers/utils/generation.rb', line 284

def initialize(max_length, forced_eos_token_id)
  super()
  @max_length = max_length
  @forced_eos_token_id = forced_eos_token_id
end

Instance Method Details

#call(input_ids, logits) ⇒ Object

# File 'lib/informers/utils/generation.rb', line 290

def call(input_ids, logits)
end