In 2nd stage training AttributeError: 'AudioDiffusionConditional' object has no attribute 'module'

[SandyPanda-MLDL]

I am unable to debug this specific issue which is appearing in 2nd stage training as follows:

Traceback (most recent call last):
File "/hdd2/Sandipan/SDhar-Projects/StyleTTS2-new/train_second.py", line 829, in
main()
File "/hdd2/Sandipan/envs/stylett2new/lib/python3.9/site-packages/click/core.py", line 1157, in call
return self.main(*args, **kwargs)
File "/hdd2/Sandipan/envs/stylett2new/lib/python3.9/site-packages/click/core.py", line 1078, in main
rv = self.invoke(ctx)
File "/hdd2/Sandipan/envs/stylett2new/lib/python3.9/site-packages/click/core.py", line 1434, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/hdd2/Sandipan/envs/stylett2new/lib/python3.9/site-packages/click/core.py", line 783, in invoke
return __callback(*args, **kwargs)
File "/hdd2/Sandipan/SDhar-Projects/StyleTTS2-new/train_second.py", line 353, in main
model.diffusion.module.diffusion.sigma_data = s_trg.std(axis=-1).mean().item() # batch-wise std estimation
File "/hdd2/Sandipan/envs/stylett2new/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1614, in getattr
raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'AudioDiffusionConditional' object has no attribute 'module'