Michael Goin 962d2c6349 [Model][Pixtral] Use memory_efficient_attention for PixtralHFVision (#9520)

2024-10-20 05:29:14 +00:00

..

guided_decoding

[Frontend][Core] Move guided decoding params into sampling params (#8252)

2024-10-01 09:34:25 +08:00

layers

[Model] Support Pixtral models in the HF Transformers format (#9036)

2024-10-18 13:29:56 -06:00

model_loader

[Hardwware][Neuron] Simplify model load for transformers-neuronx library (#9380)

2024-10-17 15:39:39 -07:00

models

[Model][Pixtral] Use memory_efficient_attention for PixtralHFVision (#9520)

2024-10-20 05:29:14 +00:00

__init__.py

[Performance] Optimize e2e overheads: Reduce python allocations (#7162)

2024-08-08 21:34:28 -07:00

custom_op.py

[torch.compile] Fine-grained CustomOp enabling mechanism (#9300)

2024-10-17 18:36:37 +00:00

parameter.py

[Kernel] (2/N) Machete - Integrate into CompressedTensorsWNA16 and GPTQMarlin (#7701)

2024-09-23 13:46:26 -04: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

[Spec Decode] (1/2) Remove batch expansion (#8839)

2024-10-01 16:04:42 -07:00

utils.py

[TPU] Call torch._sync(param) during weight loading (#9437)

2024-10-17 09:00:11 -07:00