src/transformers/models/blenderbot/modeling_tf_blenderbot.py/TFBlenderbotForConditionalGeneration/set_bias
class TFBlenderbotForConditionalGeneration:    def set_bias(self, value):
        # Replaces the existing layers containing bias for correct (de)serialization.
        vocab_size = value["final_logits_bias"].shape[-1]
        self.bias_layer = BiasLayer(
            name="final_logits_bias", shape=[1, vocab_size], initializer="zeros", trainable=False
        )
        self.bias_layer.bias.assign(value["final_logits_bias"])