chialin
|
cd2ffd7229
|
Update latte.py
revise the xformers inputs
Former-commit-id: 60db66c912192c3fbafe55ebe16157514b884e5b
|
2024-03-08 20:02:31 +08:00 |
|
LinB203
|
bc4e4f6fa6
|
del use_fp16
Former-commit-id: 5424c0112e04b56d1df30ac76cbc104e848406a3
|
2024-03-08 17:27:22 +08:00 |
|
LinB203
|
af49b2a9e9
|
init pos embed
Former-commit-id: 28f8076136ec214c66039ead7547c9b25cfe7bfc
|
2024-03-07 20:01:10 +08:00 |
|
LinB203
|
cca84f1c6c
|
denorm fun
Former-commit-id: 05707a07b14295f5e1932d0c10bb84a20a0cd328
|
2024-03-07 10:40:06 +08:00 |
|
LinB203
|
1b99cf2789
|
add sample script
Former-commit-id: 9ce847651bb6dba28c51ce5807e222ae2902a78a
|
2024-03-06 23:23:17 +08:00 |
|
LinB203
|
193d7ef7a0
|
support latte training
Former-commit-id: d471a9414842f422d1200c2fd3f3ddfcf9e97fd6
|
2024-03-06 20:29:31 +08:00 |
|
LinB203
|
5a80da7ec1
|
add vae and fix bug in dit
Former-commit-id: 2dd4254c22ce671d5cd5e7d22a2386b7660fbdc4
|
2024-03-06 19:32:20 +08:00 |
|
kabachuha
|
abffb358f9
|
support for ring attention
Former-commit-id: 0533c73b99217e88da121bd0f68e0949bc88016e
|
2024-03-06 12:24:10 +03:00 |
|
kabachuha
|
e1459ff3d0
|
option to use rebased linear attention
Former-commit-id: f1b39b34df5fbdef14ef3f6a0e6d0972a8570445
|
2024-03-06 12:18:03 +03:00 |
|
LinB203
|
cf40ce1e7a
|
add latte
Former-commit-id: 01ba10e5d65c9d2d957472fbe7958e1b07a971e9
|
2024-03-06 15:28:20 +08:00 |
|