AttributeError: 'ConvSE3' object has no attribute 'weights'

[aravinda1879]

Hi,
I tried running the first example and got the "'ConvSE3' object has no attribute 'weights'" error. I did use the SE3Transformer in this repo. Any directions are highly appreciated.
Thanks!

Command: ../run_RF2.sh rcsb_pdb_7UGF.fasta -o 7UGF

error:
Running RoseTTAFold2 to predict structures
-> Running command: python RoseTTAFold2/network/predict.py -inputs 7UGF/rcsb_pdb_7UGF_1.msa0.a3m -prefix 7UGF/models/model -model /software/RoseTTAFold2/network/weights/RF2_apr23.pt -db /software/RoseTTAFold2/pdb100_2021Mar03/pdb100_2021Mar03 -symm C1
/software/RoseTTAFold2/network/weights/RF2_apr23.pt
Running on GPU
Traceback (most recent call last):
File "/software/RoseTTAFold2/network/predict.py", line 488, in
pred = Predictor(args.model, torch.device("cuda:0"))
File "/software/RoseTTAFold2/network/predict.py", line 165, in init
self.model = RoseTTAFoldModule(
File "/software/RoseTTAFold2/network/RoseTTAFoldModel.py", line 30, in init
self.simulator = IterativeSimulator(n_extra_block=n_extra_block,
File "/software/RoseTTAFold2/network/Track_module.py", line 591, in init
self.extra_block = nn.ModuleList([IterBlock(d_msa=d_msa_full, d_pair=d_pair,
File "/software/RoseTTAFold2/network/Track_module.py", line 591, in
self.extra_block = nn.ModuleList([IterBlock(d_msa=d_msa_full, d_pair=d_pair,
File "/software/RoseTTAFold2/network/Track_module.py", line 541, in init
self.str2str = Str2Str(d_msa=d_msa, d_pair=d_pair,
File "/software/RoseTTAFold2/network/Track_module.py", line 449, in init
self.se3 = SE3TransformerWrapper(**SE3_param)
File "/software/RoseTTAFold2/network/SE3_network.py", line 58, in init
self.reset_parameter()
File "/software/RoseTTAFold2/network/SE3_network.py", line 82, in reset_parameter
nn.init.zeros_(self.se3.graph_modules[-1].weights['0'])
File "/software/RFDesign/envs/.conda/envs/rfdesign/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1130, in getattr
raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'ConvSE3' object has no attribute 'weights'