Class: Informers::WhisperForConditionalGeneration

Inherits:
WhisperPreTrainedModel show all
Defined in:
lib/informers/models.rb

Constant Summary collapse

REQUIRES_ATTENTION_MASK = 
false
MAIN_INPUT_NAME = 
:input_features

Instance Attribute Summary

Attributes inherited from PreTrainedModel

#config

Instance Method Summary collapse

Methods inherited from PreTrainedModel

#call, construct_session, from_pretrained

Constructor Details

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

Returns a new instance of WhisperForConditionalGeneration.



1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
 
# File 'lib/informers/models.rb', line 1177

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["decoder_layers"]
  @num_decoder_heads = @config["decoder_attention_heads"]
  @decoder_dim_kv = @config["d_model"] / @num_decoder_heads.to_f

  @num_encoder_layers = @config["encoder_layers"]
  @num_encoder_heads = @config["encoder_attention_heads"]
  @encoder_dim_kv = @config["d_model"] / @num_encoder_heads.to_f
end

Instance Method Details

#generate(inputs, generation_config = nil, logits_processor = nil) ⇒ Object

Raises:



1191
1192
1193
 
# File 'lib/informers/models.rb', line 1191

def generate(inputs, generation_config = nil, logits_processor = nil)
  raise Todo
end