This website requires JavaScript.
d0c53871a4
fix tensor type hint
will/update_min_python
SolitaryThinker
2025-05-18 08:33:47 +00:00
0d5306f61f
update min python to 3.10
SolitaryThinker
2025-05-18 08:12:55 +00:00
12647457a7
[Misc] Small fixes to Torch code (#395 )
applesaucethebun and Brayden Zhong
2025-05-23 17:40:24 -04:00
298f74f956
Set device for encode (#420 )
Kevin Lin
2025-05-23 16:19:45 -05:00
ee8babb298
Unify env report script in issue template (#423 )
Wenxuan Tan
2025-05-23 16:19:18 -05:00
60295cc03f
Use version.py (#424 )
Wenxuan Tan
2025-05-23 16:17:33 -05:00
1572e13b6e
[Tests] don't run 3.10 and 3.11 for SSIM (#427 )
William Lin
2025-05-23 12:59:50 -07:00
8c4704edf5
update train dependecies
Will Lin
2025-05-23 12:48:06 -07:00
dfba7ec833
Small fix
JerryZhou54
2025-05-23 19:03:58 +00:00
7a2e171f1b
Small fix
JerryZhou54
2025-05-23 19:01:59 +00:00
a8aac6090a
Integrate the new parquet dataloader into training pipeline
JerryZhou54
2025-05-23 18:27:52 +00:00
191d1be3b4
Will/training (#425 )
2025-05-23 02:06:54 -04:00
338ea1e5f2
Add script to upload preprocessed dataset to HF
JerryZhou54
2025-05-23 06:05:40 +00:00
702e77c937
edit to attention
zihang-textencoder
Zihang-He
2025-05-23 01:39:30 +00:00
42a2f272d5
Preprocessing Stage: 1. Save to parquets periodically 2. Allow resuming from the middle 3. Doesn't support multi-gpu for now. Data Loader Stage: 1. Allow multi-gpu dataloader 2. Doesn't drop files even if number of parquet files is not divisible by num_gpus 3. Able to resume training
JerryZhou54
2025-05-23 00:57:39 +00:00
a157275b4c
Fix version number (#422 )
Wenxuan Tan
2025-05-22 14:37:34 -05:00
dd9e8b9756
[Fix] drop file when num_parquet % num_workers != 0
jinzhe/fix/parquet_dataset
Jinzhe Pan
2025-05-22 21:06:46 +08:00
6f38374db2
[Fmt] fmt code
jinzhe/flux
Jinzhe Pan
2025-05-21 20:19:11 +08:00
24f30c48f1
[Fix] return type of generate_image
Jinzhe Pan
2025-05-21 20:19:11 +08:00
8c879b9bec
[Fix] return type of Work & executor
Jinzhe Pan
2025-05-21 20:12:15 +08:00
239f69052f
[Fix] add image_generate to VideoGeneartor & flux code
Jinzhe Pan
2025-05-21 18:05:03 +08:00
e28b260062
fix test
Will Lin
2025-05-06 17:56:49 -07:00
a86dca5513
ignore file in lint
Will Lin
2025-05-06 16:45:09 -07:00
05d98c6738
lint
Will Lin
2025-05-06 14:33:35 -07:00
c11920aa50
[Fix] refactor image generation related code & flux code
Jinzhe Pan
2025-05-07 04:25:50 +08:00
19b9e3d9a1
[WIP][fix] accuracy issues
Jinzhe Pan
2025-05-07 03:32:39 +08:00
177ff1e3b0
[WIP][test] add test for FluxTransformer2D
Jinzhe Pan
2025-05-06 05:40:48 +08:00
e733da23cc
[WIP][feat] support flux in v1, images were not successfully generated, still need to be fixed
Jinzhe Pan
2025-05-06 03:32:11 +08:00
3f0d052fdb
[WIP][feat] support flux in v1, finish config
Jinzhe Pan
2025-05-05 16:55:23 +08:00
9d03a5945f
[WIP][feat] support flux in v1, finish FluxTransformer2DModel
Jinzhe Pan
2025-05-04 18:51:16 +08:00
31bfea5fce
[WIP][feat] support flux in v1, adding FluxTransformer2DModel
Jinzhe Pan
2025-05-04 16:22:02 +08:00
e6a44f96bc
[WIP][feat] add flux model to v1 code
Jinzhe Pan
2025-05-01 23:12:07 +08:00
c4dbe7dac3
[bug] fix bs > 1 (#418 )
William Lin
2025-05-21 21:07:01 -07:00
d39591108e
Fulfill worker response on interrupt (#417 )
Kevin Lin
2025-05-21 22:59:11 -05:00
982bfcfdc8
Fix small issues when loading mask
JerryZhou54
2025-05-21 22:46:32 +00:00
2ac06379a7
Finish data preprocessing and loading
JerryZhou54
2025-05-21 22:34:03 +00:00
2d43f78d04
compared tokenizer
Zihang-He
2025-05-20 03:09:40 +00:00
baaa1673f7
Add data preprocessing script for WAN
JerryZhou54
2025-05-19 15:09:04 +00:00
e21a124f06
added bert encoder
Zihang-He
2025-05-19 03:23:31 +00:00
ace6e971e5
[V1] Remove vLLM dependency (#413 )
William Lin
2025-05-18 00:57:28 -07:00
b4f6758253
[Teacache] allow None for forward_context batch when using teacache (#412 )
William Lin
2025-05-17 18:43:14 -07:00
535d29b392
[Docs] Fix image (#407 )
William Lin
2025-05-12 14:30:05 -07:00
b4255517e0
[Docs] Add CLI docs (#406 )
William Lin
2025-05-12 14:16:48 -07:00
6eeb60613f
Release 0.1.0 (#405 )
v0.1.0
William Lin
2025-05-12 11:54:23 -07:00
53d2c7791f
[V1] Update where num_frame rounding is done (#403 )
William Lin
2025-05-12 11:52:49 -07:00
53cb693dca
[V1] Docs Update (#402 )
William Lin
2025-05-12 11:52:09 -07:00
6f72d24876
[CLI] Default to pipeline config (#401 )
Kevin Lin
2025-05-12 02:22:53 -05:00
d1459e9976
[V1] Update README (#400 )
William Lin
2025-05-11 22:32:03 -07:00
59ab481eb1
release 0.0.5 (#399 )
v0.0.5
William Lin
2025-05-11 16:23:11 -07:00
0cf001986a
[Docs] More docs update (#394 )
William Lin
2025-05-11 16:19:47 -07:00
51956369a5
[Misc] Replace instances of time.time() with time.perf_counter() (#396 )
applesaucethebun and Brayden Zhong
2025-05-11 18:56:39 -04:00
4b0970cbbf
[CI] Set volume_size required to false (#398 )
Kevin Lin
2025-05-11 17:55:34 -05:00
94bf47a572
[CI] Set default disk size (#397 )
Kevin Lin
2025-05-11 17:34:17 -05:00
1a3ac9074b
Zihang stepvideo v1 (#389 )
River (Zihang He) and Will Lin
2025-05-11 14:41:42 -07:00
fb0581d5b0
[CLI] Update cli to support new api/model config (#384 )
Kevin Lin
2025-05-11 15:50:33 -05:00
6c74ab4132
[CI] Use python 3.10/3.11 for SSIM test (#392 )
Kevin Lin
2025-05-10 15:48:29 -05:00
9a91021c56
[Docs] Add collect_env.py and various docs update (#393 )
William Lin
2025-05-10 13:42:42 -07:00
d8bf1beabe
fix test
will/flux-ci
Will Lin
2025-05-06 17:56:49 -07:00
30ea5bbbbc
ignore file in lint
Will Lin
2025-05-06 16:45:09 -07:00
fd80ccbf88
lint
Will Lin
2025-05-06 14:33:35 -07:00
93c30ce848
[Fix] refactor image generation related code & flux code
Jinzhe Pan
2025-05-07 04:25:50 +08:00
ea134fd785
[WIP][fix] accuracy issues
Jinzhe Pan
2025-05-07 03:32:39 +08:00
4501ef4745
[WIP][test] add test for FluxTransformer2D
Jinzhe Pan
2025-05-06 05:40:48 +08:00
979e3e8d5b
[WIP][feat] support flux in v1, images were not successfully generated, still need to be fixed
Jinzhe Pan
2025-05-06 03:32:11 +08:00
6b8f2d8228
[WIP][feat] support flux in v1, finish config
Jinzhe Pan
2025-05-05 16:55:23 +08:00
3281151955
[WIP][feat] support flux in v1, finish FluxTransformer2DModel
Jinzhe Pan
2025-05-04 18:51:16 +08:00
60f68d5b68
[WIP][feat] support flux in v1, adding FluxTransformer2DModel
Jinzhe Pan
2025-05-04 16:22:02 +08:00
d20b8df607
[WIP][feat] add flux model to v1 code
Jinzhe Pan
2025-05-01 23:12:07 +08:00
51c94d6a73
[Misc] Small Fixes & Features (#390 )
Wei Zhou
2025-05-07 18:40:49 -04:00
dba38dbc03
Release 0.0.4 (#388 )
William Lin
2025-05-07 02:40:42 -07:00
2034cc3c4f
[misc] Improve worker cleanup (#387 )
William Lin
2025-05-06 22:45:22 -07:00
c69afce2f6
[Docs] Update for V1 (#381 )
William Lin
2025-05-06 16:34:44 -07:00
b08e758eb3
Small Fixes & Features (#378 )
Wei Zhou
2025-05-06 18:52:34 -04:00
3f3462d7ce
Cleanup Teacache params (#386 )
William Lin
2025-05-06 15:52:05 -07:00
c9c47dd89c
Add Teacache to V1 (#371 )
William Lin
2025-05-06 11:58:56 -07:00
9c4ef7c2f1
[Lint] fix (#382 )
William Lin
2025-05-06 00:42:55 -07:00
f25eb4b905
[CI] Add write permissions to build-image workflow (#379 )
Kevin Lin
2025-05-05 21:21:03 -05:00
048d55ccbb
[CI] Add new images for different Python versions (#377 )
Kevin Lin
2025-05-05 12:28:33 -05:00
a271c55fe4
Fix FSDP issues when using cpu_offload flag (#376 )
Wei Zhou
2025-05-02 18:54:52 -04:00
f663ae0d8a
change gradio example to use model configs (#375 )
William Lin
2025-05-02 10:42:03 -07:00
c0911aa3dd
release 0.0.3 (#374 )
William Lin
2025-05-02 03:50:41 -07:00
5f59687ae7
Fix model config for python 3.11+ (#373 )
William Lin
2025-05-02 03:18:56 -07:00
5adbc81cdc
Refactor encoder (#370 )
Wei Zhou
2025-05-02 05:41:12 -04:00
f26d5c37c1
Update SSIM tests to use new API (#369 )
Kevin Lin
2025-05-01 06:14:24 -05:00
ab55216810
[V1] Model config (#358 ) (#368 )
model_config
Wei Zhou
2025-04-30 18:54:28 -04:00
6a4ef42378
[V1] Model config (#358 )
Wei Zhou
2025-04-30 17:48:22 -04:00
477555384e
Small fix
JerryZhou54
2025-04-30 20:55:36 +00:00
24c3ac6ea2
Add sage attn
JerryZhou54
2025-04-30 01:55:30 +00:00
d9689cef95
replace arch_config.param with config.param
JerryZhou54
2025-04-30 01:43:51 +00:00
60f01567d1
Pass pre-commit tests
JerryZhou54
2025-04-29 03:24:04 +00:00
e16222a5cc
Add Model Config for encoders
JerryZhou54
2025-04-28 00:54:38 +00:00
4e297a47e5
Add model config for DiTs
JerryZhou54
2025-04-24 22:16:44 +00:00
81bc6c9943
Separate FastVideoArgs from SamplingParam
JerryZhou54
2025-04-24 07:53:44 +00:00
80450962ef
Complete Model Config design for VAEs
JerryZhou54
2025-04-20 06:29:08 +00:00
7e6236a863
Add model config for WanVAE
JerryZhou54
2025-04-18 04:54:51 +00:00
684f7feee1
Add model config for VAE
JerryZhou54
2025-04-18 02:30:54 +00:00
c614f154e2
Add preliminary design for model config
JerryZhou54
2025-04-17 16:13:26 +00:00
f1098c77dc
[Attn] Add SageAttention Backend (#366 )
William Lin
2025-04-28 12:40:07 -06:00
0405b618f8
[Docs] Docs for design and adding new pipeline (#363 )
William Lin
2025-04-24 01:04:10 -07:00
38ee9dc3b4
Complete Model Config design for VAEs
wei/api
JerryZhou54
2025-04-20 06:29:08 +00:00