pixeli99 / SVD_Xtend

Stable Video Diffusion Training Code and Extensions.
574 stars 55 forks source link

Error when I start the training #46

Closed andreichisilev closed 6 months ago

andreichisilev commented 6 months ago

I get an error about the UNetSpatioTemporalConditionModel. This is the Traceback: Traceback (most recent call last): File "/workspace/SVD_Xtend/train_svd.py", line 1255, in main() File "/workspace/SVD_Xtend/train_svd.py", line 1043, in main added_time_ids = _get_add_time_ids( File "/workspace/SVD_Xtend/train_svd.py", line 951, in _get_add_time_ids expected_add_embed_dim = unet.module.add_embedding.linear_1.in_features File "/opt/conda/lib/python3.10/site-packages/diffusers/models/modeling_utils.py", line 218, in getattr return super().getattr(name) File "/opt/conda/lib/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: 'UNetSpatioTemporalConditionModel' object has no attribute 'module'

jhj7905 commented 6 months ago

@andreichisilev You can solve it with https://github.com/pixeli99/SVD_Xtend/issues/2

andreichisilev commented 6 months ago

Thank you