Конкатенация керасов не определяется при загрузке модели

В одном из моих лямбда-слоев я использовал from keras.layers import concatenate для объединения двух тензоров, и во время обучения она работала без проблем, и я успешно сохранил файлы модели.

Однако, когда я загружаю модель, она выдает мне эту ошибку:

NameError: имя 'concatenate' не определено

Кто-нибудь знает, что может быть не так? Я импортировал сцепление перед загрузкой модели.

Лямбда-слой выглядит так:

def concat_l1_l2(vests):
    l1, l2 = vests
    l1 = K.l2_normalize(l1, axis=-1)
    l2 = K.l2_normalize(l2, axis=-1)
    return concatenate([l1, l2])