search

pixeli99 / SVD_Xtend

Stable Video Diffusion Training Code and Extensions.
481 stars 45 forks source link

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

Closed dailingx closed 7 months ago

dailingx commented 7 months ago

I got an error when I run with: accelerate launch train_svd.py --pretrained_model_name_or_path=/root/stable-video-diffusion-img2vid-xt --output_dir="model_out" --per_gpu_batch_size=1 --gradient_accumulation_steps=1 --max_train_steps=50000 --width=512 --height=320 --checkpointing_steps=1000 --checkpoints_total_limit=1 --learning_rate=1e-5 --lr_warmup_steps=0 --seed=123 --mixed_precision="fp16" --validation_steps=200

train_dataloader is <accelerate.data_loader.DataLoaderShard object at 0x7f4c101b5150> Traceback (most recent call last): File "/root/SVD_Xtend/train_svd.py", line 1246, in main() File "/root/SVD_Xtend/train_svd.py", line 1025, in main added_time_ids = _get_add_time_ids( File "/root/SVD_Xtend/train_svd.py", line 940, in _get_add_time_ids passed_add_embed_dim = unet.module.config.addition_time_embed_dim * \ File "/root/SVD_Xtend/venv/lib/python3.10/site-packages/diffusers/models/modeling_utils.py", line 222, in getattr return super().getattr(name) File "/root/SVD_Xtend/venv/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1695, in getattr raise AttributeError(f"'{type(self).name}' object has no attribute '{name}'") AttributeError: 'UNetSpatioTemporalConditionModel' object has no attribute 'module'. Did you mean: 'modules'?

pixeli99 commented 7 months ago

hi,

Same as #2

dailingx commented 7 months ago

Thanks, this is very useful.