Zhang PeiyuanandSolitaryThinker 2325adffa2 Add STA to V1 (#312) ...

Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com>

2025-04-15 02:39:58 -07:00

..

__init__.py

[do not merge] Rebased refactor (#270)

2025-03-29 17:43:47 -05:00

activation.py

[Model] Remove RMSNorm's forward_native hardcode from Wan (#339)

2025-04-14 16:46:43 -07:00

custom_op.py

[do not merge] Rebased refactor (#270)

2025-03-29 17:43:47 -05:00

layernorm.py

Add STA to V1 (#312)

2025-04-15 02:39:58 -07:00

linear.py

[Model] Remove RMSNorm's forward_native hardcode from Wan (#339)

2025-04-14 16:46:43 -07:00

mlp.py

Add STA to V1 (#312)

2025-04-15 02:39:58 -07:00

rotary_embedding.py

[Model] Remove RMSNorm's forward_native hardcode from Wan (#339)

2025-04-14 16:46:43 -07:00

utils.py

[do not merge] Rebased refactor (#270)

2025-03-29 17:43:47 -05:00

visual_embedding.py

Add STA to V1 (#312)

2025-04-15 02:39:58 -07:00

vocab_parallel_embedding.py

[CI] Use pre-commit to run linter (#321)

2025-04-08 11:38:47 -07:00