Wenxuan Tan c5155b256e [Feature] Load weights from distributed (#470)

2025-06-27 22:52:40 -05:00

..

lora

[misc] Add missing license headers (#499)

2025-06-10 14:25:32 -07:00

quantization

[V1] Remove vLLM dependency (#413)

2025-05-18 00:57:28 -07:00

__init__.py

[do not merge] Rebased refactor (#270)

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

activation.py

Revert "Add torch.compile for all small ops" (#484)

2025-06-07 07:21:17 -05:00

custom_op.py

[V1] Remove vLLM dependency (#413)

2025-05-18 00:57:28 -07:00

layernorm.py

[Feature] Load weights from distributed (#470)

2025-06-27 22:52:40 -05:00

linear.py

[bugfix] [misc] fix denoising stage init; rename distributed env function; fix logging. (#481)

2025-06-06 20:01:10 -07:00

mlp.py

Revert "Add torch.compile for all small ops" (#484)

2025-06-07 07:21:17 -05:00

rotary_embedding.py

Revert "Add torch.compile for all small ops" (#484)

2025-06-07 07:21:17 -05:00

utils.py

[do not merge] Rebased refactor (#270)

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

visual_embedding.py

Revert "Add torch.compile for all small ops" (#484)

2025-06-07 07:21:17 -05:00

vocab_parallel_embedding.py

[bugfix] [misc] fix denoising stage init; rename distributed env function; fix logging. (#481)

2025-06-06 20:01:10 -07:00