Модуль keras.utils.generic_utils не имеет атрибута get_custom_objects при импорте segmentation_models

Я работаю над коллаборацией Google с библиотекой segmentation_models. В первую неделю он работал отлично, но теперь кажется, что я больше не могу импортировать библиотеку. Вот сообщение об ошибке, когда я выполняю import segmentation_models as sm :

---------------------------------------------------------------------------

AttributeError                            Traceback (most recent call last)

<ipython-input-3-6f48ce46383f> in <module>
      1 import tensorflow as tf
----> 2 import segmentation_models as sm

            3 frames

/usr/local/lib/python3.8/dist-packages/efficientnet/__init__.py in init_keras_custom_objects()
     69     }
     70 
---> 71     keras.utils.generic_utils.get_custom_objects().update(custom_objects)
     72 
     73 

AttributeError: module 'keras.utils.generic_utils' has no attribute 'get_custom_objects'

Colab использует версию tensorflow 2.11.0.

Я не нашел никакой информации об этом конкретном сообщении об ошибке. Кто-нибудь знает, откуда может исходить проблема?