Swapnil ParekhandSwapnil Parekh 819924e749 [Core] Adding token ranks along with logprobs (#3516) ...

Co-authored-by: Swapnil Parekh <swapnilp@ibm.com>

2024-03-25 10:13:10 -07:00

..

layers

[Core] Adding token ranks along with logprobs (#3516)

2024-03-25 10:13:10 -07:00

models

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

parallel_utils

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

guided_decoding.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

guided_logits_processors.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

model_loader.py

[Neuron] Support inference with transformers-neuronx (#2569)

2024-02-28 09:34:34 -08:00

neuron_model_loader.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

sampling_metadata.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

utils.py

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

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

weight_utils.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00