problems with loading the checkpoints for planar.ckpt?

[xinyue96]

Hi,

I am having some trouble loading the checkpoint file (e.g. checkpoint_planar.ckpt) I cloned the latest version of your code and did: from diffusion_model_discrete import DiscreteDenoisingDiffusion model =DiscreteDenoisingDiffusion.load_from_checkpoint('/.../checkpoint_planar.ckpt',map_location='cuda:0')

Then I get the error AttributeError: module 'datasets.spectre_dataset' has no attribute 'PlanarDataModule'

I then went to an earlier version of your code where PlanarDataModule is defined in datasets.spectre_dataset, but then I encounter the error: AttributeError: module 'pytorch_lightning.utilities.enums' has no attribute '_FaultTolerantMode'

I then upgraded pytorch_lightning from 2.0.4 to 2.2.1, but still get a new error of: AttributeError: module 'pytorch_lightning.trainer.connectors.data_connector' has no attribute 'DataConnector'

Would be greatly appreciated if anyone can tell me how to correctly load the provided ckpt. Thanks!