failed when repeating the tutorial

[gvenus]

Hi,

Thanks for providing us with the Matgl code.

I failed to repeat the online tutorial (https://matgl.ai/tutorials%2FTraining%20a%20M3GNet%20Potential%20with%20PyTorch%20Lightning.html). The error is:
(matgl) [ ]$ GPU available: False, used: False TPU available: False, using: 0 TPU cores HPU available: False, using: 0 HPUs Traceback (most recent call last): File "1-her-ml/1-ML-POT-traning.py", line 177, in <module> trainer.fit(model=lit_module_finetune, train_dataloaders=train_loader, val_dataloaders=val_loader) File "miniconda3/envs/matgl/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 532, in fit model = _maybe_unwrap_optimized(model) File "miniconda3/envs/matgl/lib/python3.10/site-packages/pytorch_lightning/utilities/compile.py", line 111, in _maybe_unwrap_optimized raise TypeError( TypeError: modelmust be aLightningModuleortorch._dynamo.OptimizedModule, got PotentialLightningModule`

Any suggestions?

Best