Peiyuan Zhang b628b5fc7e save

2025-12-23 01:31:21 +00:00

..

attention

save

2025-12-23 01:31:21 +00:00

configs

[feat]: add COSMOS 2.5 DiT implementation (#897)

2025-12-07 21:48:32 -08:00

dataset

Add wan2.1 functionality support for Ascend NPU platform (#810)

2025-10-09 16:25:08 -07:00

distributed

[feat] Support sequence packing and shard after pachification for USP (#894)

2025-12-19 16:19:46 -06:00

entrypoints

[feat] Support sequence packing and shard after pachification for USP (#894)

2025-12-19 16:19:46 -06:00

layers

update

2025-12-23 01:28:08 +00:00

logging_utils

[Feature] Remove V1 folder (#642)

2025-07-24 22:43:12 -07:00

models

[feat] Support sequence packing and shard after pachification for USP (#894)

2025-12-19 16:19:46 -06:00

pipelines

[feat] Support sequence packing and shard after pachification for USP (#894)

2025-12-19 16:19:46 -06:00

platforms

+ generator sage 3

2025-12-23 01:21:59 +00:00

tests

update

2025-12-23 01:28:09 +00:00

third_party

[Feature] Remove V1 folder (#642)

2025-07-24 22:43:12 -07:00

training

save

2025-12-23 01:31:21 +00:00

worker

fix: correct mp backend GPU assignment on multi-GPU systems (#912)

2025-11-30 23:00:22 -08:00

workflow

[preprocessing] [self-forcing] [2/n] Improve preprocessing and add ode trajectory dataset schema (#794)

2025-09-10 17:33:57 -07:00

__init__.py

Use assert_close in tests (#429)

2025-12-06 18:18:25 -08:00

envs.py

[misc] add schedule configurations to pytorch profiler (#934)

2025-12-18 01:45:23 -06:00

fastvideo_args.py

update

2025-12-23 01:27:22 +00:00

forward_context.py

[Feature] Remove V1 folder (#642)

2025-07-24 22:43:12 -07:00

image_processor.py

[Feature] Add Cosmos2 i2v pipeline (#837)

2025-10-30 20:03:57 -07:00

logger.py

[misc] [docs] Various fixes for logging and docs (#758)

2025-08-23 21:13:50 -07:00

profiler.py

[misc] add schedule configurations to pytorch profiler (#934)

2025-12-18 01:45:23 -06:00

utils.py

[bugfix] always force spawn instead of fork (#852)

2025-10-23 16:36:50 -07:00

version.py

[chore] Release 0.1.6 (#768)

2025-08-28 20:56:21 -07:00