PR #10340 [Sana bug] bug fix for 2K model config

[Sana bug] bug fix for 2K model config #10340

DN6 merged 5 commits into huggingface:main from lawrence-cj:Sana-PE

fix the Positinoal Embedding bug in 2K model;

459628c9

Change the default model to the BF16 one for more stable training and…

6bf67e7e

yiyixuxu approved these changes on 2024-12-22

make style

b32bf00b

substract buffer size

13c59548

add compute_module_persistent_sizes

574fe743

yiyixuxu commented on 2024-12-23

yiyixuxu requested a review from

DN6 1 year ago

DN6 approved these changes on 2024-12-23

DN6 merged b58868e6 into main 1 year ago

lawrence-cj deleted the Sana-PE branch 1 year ago

Reviewers

DN6

yiyixuxu

Assignees

No one assigned

Labels

None yet

Milestone

No milestone