Method: Informers::T5ForConditionalGeneration#initialize

Defined in:
lib/informers/models.rb

#initialize(config, session, decoder_merged_session, generation_config) ⇒ T5ForConditionalGeneration

Returns a new instance of T5ForConditionalGeneration.



829
830
831
832
833
834
835
836
837
838
839
840
841
 
# File 'lib/informers/models.rb', line 829

def initialize(config, session, decoder_merged_session, generation_config)
  super(config, session)
  @decoder_merged_session = decoder_merged_session
  @generation_config = generation_config

  @num_decoder_layers = @config[:num_decoder_layers]
  @num_decoder_heads = @config[:num_heads]
  @decoder_dim_kv = @config[:d_kv]

  @num_encoder_layers = @config[:num_layers]
  @num_encoder_heads = @config[:num_heads]
  @encoder_dim_kv = @config[:d_kv]
end