Method: Informers::Utils::MinLengthLogitsProcessor#initialize

Defined in:
lib/informers/utils/generation.rb

#initialize(min_length, eos_token_id) ⇒ MinLengthLogitsProcessor

Returns a new instance of MinLengthLogitsProcessor.



251
252
253
254
255
 
# File 'lib/informers/utils/generation.rb', line 251

def initialize(min_length, eos_token_id)
  super()
  @min_length = min_length
  @eos_token_id = eos_token_id.is_a?(Array) ? eos_token_id : [eos_token_id]
end