search

hpcaitech / Open-Sora

Open-Sora: Democratizing Efficient Video Production for All
https://hpcaitech.github.io/Open-Sora/
Apache License 2.0
20.23k stars 1.91k forks source link

[rank0]: TypeError: __init__() got an unexpected keyword argument 'freeze_vae_2d' #498

Closed tau-yihouxiang closed 2 weeks ago

tau-yihouxiang commented 2 weeks ago

When I train the vae code in stage 1: 

    torchrun --nnodes=1 --nproc_per_node=1 scripts/train_vae.py configs/vae/train/stage1.py --data-path xxxx,

there is an error: [rank0]: TypeError: init() got an unexpected keyword argument 'freeze_vae_2d'.

I do not know the reason. Is your stage1.py's config right or not. Thank you for your contribution and fix the possible bug.

tau-yihouxiang commented 2 weeks ago

I am sure the state1/2/3.py files are wrong version. Could you please check and run the code before releasing? Thank you in advance! @zhengzangw

zhengzangw commented 2 weeks ago

@Shen-Chenhui Could you have a look of this?