def METHOD_NAME(self, state_dict):
    """
    Override to account for weights used for image features.
    """
    for tensor in ['dummy_image_enc', 'ones_mask']:
        key = f'encoder_ctxt.{tensor}'
        if hasattr(self.model.encoder_ctxt, tensor) and key not in state_dict:
            state_dict[key] = getattr(self.model.encoder_ctxt, tensor)
    if hasattr(self.model.encoder_ctxt, 'image_encoder'):
        for layer_idx, layer in enumerate(self.model.encoder_ctxt.image_encoder):
            for tensor in ['weight', 'bias']:
                key = f'encoder_ctxt.image_encoder.{layer_idx}.{tensor}'
                if hasattr(layer, tensor) and key not in state_dict:
                    state_dict[key] = getattr(layer, tensor)
    super().METHOD_NAME(state_dict)