save unet to state_dict during stage 1 training

guoqincode / Open-AnimateAnyone

Unofficial Implementation of Animate Anyone

2.9k stars 233 forks source link

Closed qiuyanxin closed 9 months ago

guoqincode commented 9 months ago

In the latest train.py, I did not save unet because it is frozen.