MCPcopy Create free account
hub / github.com/OpenNMT/CTranslate2 / set_decoder

Method set_decoder

python/ctranslate2/converters/transformers.py:585–593  ·  view source on GitHub ↗
(self, spec, decoder)

Source from the content-addressed store, hash-verified

583 config.unk_token = tokenizer.unk_token
584
585 def set_decoder(self, spec, decoder):
586 super().set_decoder(spec, decoder)
587
588 if decoder.project_in is not None:
589 self.set_linear(spec.project_in, decoder.project_in)
590 if decoder.project_out is not None:
591 self.set_linear(spec.project_out, decoder.project_out)
592 if decoder.final_layer_norm is not None:
593 self.set_layer_norm(spec.layer_norm, decoder.final_layer_norm)
594
595 def set_common_layers(self, spec, module):
596 spec.scale_embeddings = False

Callers 1

get_model_specMethod · 0.95

Calls 3

set_decoderMethod · 0.45
set_linearMethod · 0.45
set_layer_normMethod · 0.45

Tested by

no test coverage detected