Awni Hannun 8b05bb6d18 [mlx-lm] Use sdpa in llama / mistral model (#515) ...

* use sdpa

* update a few more models

* version

* fix stablelm type

2024-03-07 17:41:23 -08:00

..

__init__.py

Mlx llm package (#301)

2024-01-12 10:25:56 -08:00

base.py

Mlx llm package (#301)

2024-01-12 10:25:56 -08:00

gemma.py

[mlx-lm] Use sdpa in llama / mistral model (#515)

2024-03-07 17:41:23 -08:00

layers.py

[mlx-lm] Add precompiled normalizations (#451)

2024-02-22 12:40:55 -08:00

llama.py

[mlx-lm] Use sdpa in llama / mistral model (#515)

2024-03-07 17:41:23 -08:00

mixtral.py

Enable unit testing in Circle and start some MLX LM tests (#545)

2024-03-07 09:31:57 -08:00

olmo.py

Enable unit testing in Circle and start some MLX LM tests (#545)

2024-03-07 09:31:57 -08:00

phi.py

Enable unit testing in Circle and start some MLX LM tests (#545)

2024-03-07 09:31:57 -08:00

phixtral.py

[mlx-lm] Add precompiled normalizations (#451)

2024-02-22 12:40:55 -08:00

plamo.py

Enable unit testing in Circle and start some MLX LM tests (#545)

2024-03-07 09:31:57 -08:00

qwen2.py

[mlx-lm] Use sdpa in llama / mistral model (#515)

2024-03-07 17:41:23 -08:00

qwen.py

Enable unit testing in Circle and start some MLX LM tests (#545)

2024-03-07 09:31:57 -08:00

stablelm.py

[mlx-lm] Use sdpa in llama / mistral model (#515)

2024-03-07 17:41:23 -08:00

starcoder2.py

[mlx-lm] Use sdpa in llama / mistral model (#515)

2024-03-07 17:41:23 -08:00