ImportError: cannot import name 'FLAX_WEIGHTS_NAME' from 'transformers.utils'

Hi guys!
I try to import DDPMScheduler, UNet2DModel from diffusers and receive this error:

ImportError: cannot import name 'FLAX_WEIGHTS_NAME' from 'transformers.utils' (/opt/python/envs/default/lib/python3.8/site-packages/transformers/utils/__init__.py)

I also tried to make this import in Colab and it works fine. How to fix this in DataLore?