LinB203
|
45187a0d22
|
fix training bug
Former-commit-id: 09a6283ddd9ab11479f30324de788bd7415702ae
|
2024-03-28 19:07:46 +08:00 |
|
qqingzheng
|
ec81e2427c
|
add causalvae
Former-commit-id: ac936276fd3213ee1815dfe096cd64cecb601428
|
2024-03-27 12:49:30 +00:00 |
|
LinB203
|
6c12429fd2
|
train t2v feature
Former-commit-id: 7f16b162cae0e3533c8ba9aa03186a846d9e02bf
|
2024-03-25 16:25:01 +08:00 |
|
LinB203
|
6b51d0ccfd
|
support attention_mode
Former-commit-id: c72f354fbfd948e8cd4b0cc63c157bc4688c7665
|
2024-03-20 22:23:24 +08:00 |
|
LinB203
|
ff36d291d8
|
train with image
Former-commit-id: 209b9a8ad2f19a50e46f9b7ec68dd781898e491a
|
2024-03-19 23:21:57 +08:00 |
|
qqingzheng
|
b3c17829e9
|
[feat] add casual vqvae
Former-commit-id: 89087970b267908c8667984aba42b1ade2505467
|
2024-03-17 02:46:33 +08:00 |
|
lb203
|
73161c3248
|
Update train_t2v.py
Former-commit-id: d25b39ffac4474abd40b551fe0e51c421cb7bf5e
|
2024-03-16 22:23:36 +08:00 |
|
lb203
|
1d1a6a5b63
|
Update train.py
Former-commit-id: 783faf5cd86c148a713ea986ceed8527b108f2b8
|
2024-03-16 22:23:22 +08:00 |
|
LinB203
|
085062a53f
|
refactor and fix resume bug
Former-commit-id: 1b1136671b9a7e0b15b70168e2c44dbff04a22e2
|
2024-03-16 22:21:15 +08:00 |
|
LinB203
|
516082e06c
|
resume and compress kv
Former-commit-id: 0af7d58f88c1b65b4089da6f3e5c70456614da7c
|
2024-03-15 22:48:09 +08:00 |
|
LinB203
|
d2a8e64a83
|
del t2v
Former-commit-id: 5daa2952997a29e74e2b5d235001d6d9dabf9ff6
|
2024-03-15 18:43:44 +08:00 |
|
LinB203
|
99a789bd5d
|
del t2v
Former-commit-id: 910398ca297be8a188d59c59c121857666774af1
|
2024-03-14 08:51:10 +08:00 |
|
LinB203
|
0e2a86b2ac
|
clean dit
Former-commit-id: 117ec4196a6f2cd83da223b993b14e28f7a693f8
|
2024-03-11 15:44:53 +08:00 |
|
sennnnn
|
1365640065
|
Fix resolution typos.
Former-commit-id: 3ad55d28eaf13f9c8fd23912d2691965737527e9
|
2024-03-11 14:26:05 +08:00 |
|
LinB203
|
b57709108a
|
train 1080p video
Former-commit-id: 74625142ef892f9e7fa459b2f98f2f7ae5b36d6a
|
2024-03-10 18:08:23 +08:00 |
|
LinB203
|
6dae0b1c0c
|
deepspeed
Former-commit-id: 6179a4b0e7d06caed4201a27679abc64615febe6
|
2024-03-10 15:08:12 +08:00 |
|
sennnnn
|
ee3810a585
|
Decouple mixed_precision and gradient_accumulation to command args.
Former-commit-id: 903d473d1432eb492a87efe184d03439f643b92d
|
2024-03-10 11:24:39 +08:00 |
|
LinB203
|
6df13b07bb
|
flash-attn train and sample
Former-commit-id: e336565bd5ff2aaf4d0b4ad5f42732e89cd216b3
|
2024-03-09 16:30:09 +08:00 |
|
LinB203
|
b0a9e72008
|
extract script
Former-commit-id: 4139f2f68ae09695302ce49afdb7318df83cabfd
|
2024-03-09 01:03:05 +08:00 |
|
LinB203
|
bc4e4f6fa6
|
del use_fp16
Former-commit-id: 5424c0112e04b56d1df30ac76cbc104e848406a3
|
2024-03-08 17:27:22 +08:00 |
|
LinB203
|
eaee962fa9
|
update train.py
Former-commit-id: 844173711342f2dcc021294e87971ca39854bfe4
|
2024-03-08 17:25:48 +08:00 |
|
Liu Hanchen
|
2448aa7c8c
|
commit
Former-commit-id: 81359cf67248b6c594ff79f121a8a38558c1cc70
|
2024-03-08 01:04:16 +08:00 |
|
LinB203
|
9c8b50a29a
|
train script
Former-commit-id: 3fd1ccf099c2deed20af50bf9556e2fc3b7a4382
|
2024-03-07 22:17:39 +08:00 |
|
LinB203
|
af49b2a9e9
|
init pos embed
Former-commit-id: 28f8076136ec214c66039ead7547c9b25cfe7bfc
|
2024-03-07 20:01:10 +08:00 |
|
LinB203
|
c521999f80
|
attn_mask with bf16
Former-commit-id: d9667a8c15a5344e3c777197fd6badec15dcddbd
|
2024-03-07 13:35:01 +08:00 |
|
Chestnut
|
589da07fec
|
Merge branch 'main' into videogpt
Former-commit-id: 714cc434cea5823fc09ef6fe502b1fe83170cae2
|
2024-03-07 01:25:58 +08:00 |
|
qqingzheng
|
b409ef399a
|
[refactor] added more methods in abstract classes
Former-commit-id: a535e3a06eb766bebe7baa3d9df64b04d04d40e6
|
2024-03-06 17:23:40 +00:00 |
|
LinB203
|
38bf4afdfa
|
support accelerate training
Former-commit-id: b32671c9991b9a9db66cb148822ccb3e8be7dc11
|
2024-03-07 00:55:35 +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 |
|
LinB203
|
cf40ce1e7a
|
add latte
Former-commit-id: 01ba10e5d65c9d2d957472fbe7958e1b07a971e9
|
2024-03-06 15:28:20 +08:00 |
|