Method: Transformers::PreTrainedModel#_initialize_weights

Defined in:
lib/transformers/modeling_utils.rb

#_initialize_weights(mod) ⇒ Object



192
193
194
195
196
197
198
# File 'lib/transformers/modeling_utils.rb', line 192

def _initialize_weights(mod)
  if mod.instance_variable_defined?(:@is_hf_initialized)
    return
  end
  _init_weights(mod)
  mod.instance_variable_set(:@is_hf_initialized, true)
end