AttributeError: 'DataModuleFromConfig' object has no attribute '_has_setup_TrainerFn.FITTING'

[shazra-raza]

I changed the requirements.txt to:
--extra-index-url https://download.pytorch.org/whl/cu113
torch==2.0
torchvision==0.15.1
albumentations>=0.4.3
opencv-python==4.5.5.64
pudb==2019.2
imageio==2.9.0
imageio-ffmpeg==0.4.2
pytorch-lightning==1.4.2
omegaconf==2.1.1
test-tube>=0.7.5
streamlit>=0.73.1
einops==0.3.0
torch-fidelity==0.3.0
transformers>=4.22.2
kornia==0.6
uvicorn==0.23.2
webdataset==0.2.5
torchmetrics==0.6.0
fire==0.4.0
dill>=0.3.6
gradio==3.1.4
diffusers==0.3.0
datasets[vision]>=2.4.0
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
-e git+https://github.com/openai/CLIP.git@main#egg=clip
-e git+https://github.com/justinpinkney/nomi.git@e9ded23b7e2269cc64d39683e1bf3c0319f552ab#egg=nomi
-e .

and now every time I run python main.py -t --base configs/stable-diffusion/pokemon.yaml --gpus="0," --scale_lr False --num_nodes 1 --check_val_every_n_epoch 10 --finetune_from /scratch/gpfs/sraza/stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt

I get the error message:

Summoning checkpoint.
Traceback (most recent call last):
File "/scratch/gpfs/sraza/fine-tuning/stable-diffusion/main.py", line 905, in
trainer.fit(model, data)
File "/home/sraza/.local/lib/python3.11/site-packages/pytorch_lightning/trainer/trainer.py", line 553, in fit
self._run(model)
File "/home/sraza/.local/lib/python3.11/site-packages/pytorch_lightning/trainer/trainer.py", line 865, in _run
self._call_setup_hook(model) # allow user to setup lightning_module in accelerator environment
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/sraza/.local/lib/python3.11/site-packages/pytorch_lightning/trainer/trainer.py", line 1169, in _call_setup_hook
self.datamodule.setup(stage=fn)
File "/home/sraza/.local/lib/python3.11/site-packages/pytorch_lightning/core/datamodule.py", line 415, in wrapped_fn
has_run = getattr(obj, attr)
^^^^^^^^^^^^^^^^^^
AttributeError: 'DataModuleFromConfig' object has no attribute '_has_setup_TrainerFn.FITTING'

Does anyone know why this could be happening?