训练lora时报错

[fallbernana123456]

在V100上，由于只有1个GPU，修改配置文件use_ddp: False，运行 python train/train_personalized.py configs/training/lora_personalization.yaml
报错
File "UltraPixel-main/train/train_personalized.py", line 368, in setup_optimizers params += list(models.train_lora.module.parameters()) File "python3.10/site-packages/torch/nn/modules/module.py", line 1688, in __getattr__ raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'") AttributeError: 'ModuleList' object has no attribute 'module'

configs/training/lora_personalization.yaml 看这里训练时不需要其他的模型是吧？

[catcathh]

目前训练代码只支持多卡的设定，如果要单卡训练，可能需要修改模型load的部分