Can't import RFlowScheduler from monai

When I try to train LDM using scripts/diff_model_train.py, It report can't find RFlowScheduler from monai.
The version of monai in my env is 1.4.0 and I saw monai release 1.4.1rc1 so I try to pip install -U monai and pip install monai-weekly but it can only install the version 1.4.0.
At last I try to install from source using BUILD_MONAI=1 pip install --no-build-isolation git+https://github.com/Project-MONAI/MONAI#egg=monai and then my computer crashed during the compilation process.

@MaybeRichard

Hi @MaybeRichard, thanks for your interest here.

Please try install the latest monai-weekly which should include the RFlowScheduler

pip install monai-weekly==1.5.dev2511

>>> import monai
>>> monai.__version__
'1.5.dev2511'
>>> from monai.networks.schedulers import RFlowScheduler

Please try uninstall the existing version first.

Hope it helps, thanks.

Navigation Menu