Latte training bug fix: temp_embed initialization

mindspore-lab / mindone

one for all, Optimal generator with No Exception

Apache License 2.0

330 stars 63 forks source link

Closed wtomin closed 3 months ago

wtomin commented 3 months ago

Fix the bug of Latte model initialization: failed to initialize temporal position embedding.