At line 66 of VAR_finetune.py, the var_wo_ddp was initialized twice, here the correct model to be initialized should be ref_var_wo_ddp.
Though this typo, it does not affect the final results, because model weights were loaded into ref_var_wo_ddp right after it was returned by the build_vae_var_with_ref().
Thanks for your effort in this work! :)
At line 66 of VAR_finetune.py, the var_wo_ddp was initialized twice, here the correct model to be initialized should be ref_var_wo_ddp.
Though this typo, it does not affect the final results, because model weights were loaded into ref_var_wo_ddp right after it was returned by the build_vae_var_with_ref().
Thanks for your effort in this work! :)