Wushi Dong 1d094fd7c0 [Distributed][PP] only create embedding & lm head when necessary (#6455) ...

original title: [Distributed][Model] Rank-based Component Creation for Pipeline Parallelism Memory Optimization

2024-07-16 19:20:26 -07:00

..

guided_decoding

[Bugfix] use diskcache in outlines _get_guide #5436 (#6203)

2024-07-08 11:23:24 -07:00

layers

[Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00

model_loader

[misc][distributed] add seed to dummy weights (#6491)

2024-07-16 19:16:34 -07:00

models

[Distributed][PP] only create embedding & lm head when necessary (#6455)

2024-07-16 19:20:26 -07:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

2024-03-25 04:39:33 +00:00

custom_op.py

[Hardware][Intel GPU] Add Intel GPU(XPU) inference backend (#3814)

2024-06-17 11:01:25 -07:00

pooling_metadata.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

sampling_metadata.py

[Core] Use numpy to speed up padded token processing (#6442)

2024-07-16 08:13:25 -07:00

utils.py

[Hardware][Neuron] Refactor neuron support (#3471)

2024-03-22 01:22:17 +00:00