Giters

pixeli99 / SVD_Xtend

Stable Video Diffusion Training Code and Extensions.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AttributeError: 'UNetSpatioTemporalConditionModel' object has no attribute 'module'. Did you mean: 'modules'?

BlakeOne opened this issue · comments

commented

Hey there, I'm trying out your new training script but I get this error... I'm using stabilityai/stable-video-diffusion-img2vid for the pretrained_model_name_or_path BTW but I'm not sure if that's correct?

AttributeError: 'UNetSpatioTemporalConditionModel' object has no attribute 'module'. Did you mean: 'modules'?

commented

SVD_Xtend/train_svd.py

Line 940 in 64c9783

passed_add_embed_dim = unet.module.config.addition_time_embed_dim * \

change to

passed_add_embed_dim = unet.config.addition_time_embed_dim * len(add_time_ids)
expected_add_embed_dim = unet.add_embedding.linear_1.in_features

You just need to delete these two lines of modules, it's due to the unfriendliness on my code because I'm using multi-GPU training, and this problem occurs under ddp mode. If you are using a single GPU, you can simply delete them.

commented

Thanks removing those lines fixed it for me. :)